New-PI credit is can now be fetched through nerc_rates

process_report/process_report.py

@@ -199,6 +199,12 @@ def main():
         required=False,
         help="Name of csv file listing all prepay group debits. If not provided, defaults to fetching from S3",
     )
+    parser.add_argument(
+        "--new-pi-credit-amount",
+        required=False,
+        type=int,
+        help="Amount of credit given to new PIs. If not provided, defaults to fetching from nerc-rates",
+    )
     parser.add_argument(
         "--BU-subsidy-amount",
         required=False,
@@ -216,6 +222,9 @@ def main():
     prepay_debits_filepath = args.prepay_debits or util.fetch_s3(
         PREPAY_DEBITS_S3_FILEPATH
     )
+    new_pi_credit_amount = args.new_pi_credit_amount or int(
+        util.fetch_nerc_rates("New PI Credit", invoice_month)
+    )
     bu_subsidy_amount = args.BU_subsidy_amount or int(
         util.fetch_nerc_rates("BU Subsidy", invoice_month)
     )
@@ -269,6 +278,7 @@ def main():
         invoice_month,
         data=validate_billable_pi_proc.data,
         old_pi_filepath=old_pi_file,
+        initial_credit_amount=new_pi_credit_amount,
         limit_new_pi_credit_to_partners=(
             util.fetch_nerc_rates(
                 "Limit New PI Credit to MGHPCC Partners", invoice_month

process_report/processors/new_pi_credit_processor.py

@@ -22,11 +22,11 @@ class NewPICreditProcessor(discount_processor.DiscountProcessor):
     """
 
     NEW_PI_CREDIT_CODE = "0002"
-    INITIAL_CREDIT_AMOUNT = 1000
     EXCLUDE_SU_TYPES = ["OpenShift GPUA100SXM4", "OpenStack GPUA100SXM4"]
     IS_DISCOUNT_BY_NERC = True
 
     old_pi_filepath: str
+    initial_credit_amount: int
     limit_new_pi_credit_to_partners: bool = False
 
     @staticmethod
@@ -117,7 +117,7 @@ def get_initial_credit_amount(
             return new_pi_credit_amount
 
         new_pi_credit_amount = get_initial_credit_amount(
-            old_pi_df, self.invoice_month, self.INITIAL_CREDIT_AMOUNT
+            old_pi_df, self.invoice_month, self.initial_credit_amount
         )
         logger.info(
             f"New PI Credit set at {new_pi_credit_amount} for {self.invoice_month}"