Skip to content

Commit 75c3b52

Browse files
MaxGhenisclaude
andcommitted
Clean up file names - remove 'final' and 'full' references
- Renamed files to use clearer, simpler names - Moved results CSV to data/NJ directory - Updated script to output to renamed file 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
1 parent 0f6ca3c commit 75c3b52

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed
File renamed without changes.
File renamed without changes.

data/NJ/nj_obbba_full_optimized.py renamed to data/NJ/nj_obbba_optimized.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212
from policyengine_us import Microsimulation
1313
from policyengine_core.reforms import Reform
1414

15-
def create_full_obbba_reform():
16-
"""Full OBBBA reform exactly as in obbba.ipynb"""
15+
def create_obbba_reform():
16+
"""OBBBA reform exactly as in obbba.ipynb"""
1717
return Reform.from_dict({
1818
# Estate tax changes
1919
"gov.irs.credits.estate.base": {
@@ -207,7 +207,7 @@ def calculate_nj_only(sim, period=2026):
207207

208208
def main():
209209
print("=" * 70)
210-
print("NJ WINNERS/LOSERS WITH FULL OBBBA REFORM")
210+
print("NJ WINNERS/LOSERS WITH OBBBA REFORM")
211211
print("Optimized for better hardware")
212212
print("=" * 70)
213213

@@ -216,7 +216,7 @@ def main():
216216

217217
print("\nThis script will:")
218218
print("1. Calculate baseline household_net_income for NJ")
219-
print("2. Apply full OBBBA reform")
219+
print("2. Apply OBBBA reform")
220220
print("3. Calculate reformed household_net_income for NJ")
221221
print("4. Analyze winners and losers by district")
222222

@@ -242,7 +242,7 @@ def main():
242242
print("-" * 70)
243243
start_reform = time.time()
244244

245-
reform = create_full_obbba_reform()
245+
reform = create_obbba_reform()
246246
sim_reform = setup_simulation(dataset_path, reform=reform)
247247
reform_income, _, _ = calculate_nj_only(sim_reform, period)
248248

@@ -329,10 +329,10 @@ def main():
329329

330330
# Save results
331331
results_df = pd.DataFrame(results)
332-
results_df.to_csv('nj_obbba_full_results.csv', index=False)
332+
results_df.to_csv('nj_obbba_results.csv', index=False)
333333

334334
print("\n" + "=" * 70)
335-
print("Results saved to nj_obbba_full_results.csv")
335+
print("Results saved to nj_obbba_results.csv")
336336
print(f"Total runtime: {time.time()-start_baseline:.1f}s")
337337
print("=" * 70)
338338

File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)