@@ -200,20 +200,18 @@ def main():
200200 logger .info (f"Using ColdFront data file at { coldfront_data_file } ." )
201201
202202 if args .use_nerc_rates :
203+
204+ def get_decimal_rate (rate_name ):
205+ return Decimal (nerc_repo_rates .get_value_at (rate_name , args .invoice_month ))
206+
203207 nerc_repo_rates = load_from_url ()
204208 rates = billing .Rates (
205- cpu = nerc_repo_rates .get_value_at ("CPU SU Rate" , args .invoice_month ),
206- gpu_a100sxm4 = nerc_repo_rates .get_value_at (
207- "GPUA100SXM4 SU Rate" , args .invoice_month
208- ),
209- gpu_a100 = nerc_repo_rates .get_value_at (
210- "GPUA100 SU Rate" , args .invoice_month
211- ),
212- gpu_v100 = nerc_repo_rates .get_value_at (
213- "GPUV100 SU Rate" , args .invoice_month
214- ),
215- gpu_k80 = nerc_repo_rates .get_value_at ("GPUK80 SU Rate" , args .invoice_month ),
216- gpu_a2 = nerc_repo_rates .get_value_at ("GPUA2 SU Rate" , args .invoice_month ),
209+ cpu = get_decimal_rate ("CPU SU Rate" ),
210+ gpu_a100sxm4 = get_decimal_rate ("GPUA100SXM4 SU Rate" ),
211+ gpu_a100 = get_decimal_rate ("GPUA100 SU Rate" ),
212+ gpu_v100 = get_decimal_rate ("GPUV100 SU Rate" ),
213+ gpu_k80 = get_decimal_rate ("GPUK80 SU Rate" ),
214+ gpu_a2 = get_decimal_rate ("GPUA2 SU Rate" ),
217215 include_stopped_runtime = (
218216 nerc_repo_rates .get_value_at (
219217 "Charge for Stopped Instances" , args .invoice_month
@@ -232,6 +230,8 @@ def main():
232230 include_stopped_runtime = args .include_stopped_runtime ,
233231 )
234232
233+ logger .info (f"Using rates: { rates } ." )
234+
235235 billing .generate_billing (
236236 args .start ,
237237 args .end ,
0 commit comments