Fix CLI tool weights file handling

PSLmodels · Sep 26, 2024 · 1d1b6e2 · 1d1b6e2
1 parent 56dba1b
commit 1d1b6e2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/taxcalc/taxcalcio.py b/taxcalc/taxcalcio.py
@@ -348,23 +348,24 @@ def init(self, input_data, tax_year, baseline, reform, assump,
                     exact_calculations=exact_calculations
                 )
             elif self.tmd_input_data:
+                wghts = pd.read_csv(self.tmd_weights)
                 recs = Records(
                     data=pd.read_csv(input_data),
                     start_year=Records.TMDCSV_YEAR,
-                    weights=self.tmd_weights,
+                    weights=wghts,
                     gfactors=gfactors_ref,
                     adjust_ratios=None,
                     exact_calculations=exact_calculations
                 )  # pragma: no cover
                 recs_base = Records(
                     data=pd.read_csv(input_data),
                     start_year=Records.TMDCSV_YEAR,
-                    weights=self.tmd_weights,
+                    weights=wghts,
                     gfactors=gfactors_base,
                     adjust_ratios=None,
                     exact_calculations=exact_calculations
                 )  # pragma: no cover
-            else:  # if not {cps|tmd}_input_data but aging_input_data
+            else:  # if not {cps|tmd}_input_data but aging_input_data: puf
                 recs = Records(
                     data=input_data,
                     gfactors=gfactors_ref,