Skip to content

Commit

Permalink
Fix specification of zeros array
Browse files Browse the repository at this point in the history
  • Loading branch information
martinholmer committed Aug 22, 2024
1 parent 751d6cb commit 00f2794
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tax_microdata_benchmarking/datasets/taxcalc_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,6 @@ def pe(variable):
"f2441": "count_cdcc_eligible",
}
# specify Tax-Calculator names of variables that have zero values
zeros = np.zeros_like(pe_dataset.tax_unit_weight)
zero_names = [
"a_lineno", # taxdata-specific (CPS matched person ID)
"agi_bin", # taxdata-specific (AGI bin)
Expand Down Expand Up @@ -118,6 +117,7 @@ def pe(variable):
var = {}
for tcname, pename in vnames.items():
var[tcname] = pe(pename)
zeros = np.zeros_like(var["RECID"])
for tcname in zero_names:
var[tcname] = zeros

Expand Down

0 comments on commit 00f2794

Please sign in to comment.