Ensure Extended FRS generates

PolicyEngine · Sep 17, 2024 · c40523c · c40523c
1 parent 8276a00
commit c40523c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -24,6 +24,7 @@ documentation:
 data:
 	python policyengine_uk_data/datasets/frs/dwp_frs.py
 	python policyengine_uk_data/datasets/frs/frs.py
+	python policyengine_uk_data/datasets/frs/extended_frs.py
 	python policyengine_uk_data/datasets/frs/enhanced_frs.py
 
 build:

diff --git a/policyengine_uk_data/utils/imputations/capital_gains.py b/policyengine_uk_data/utils/imputations/capital_gains.py
@@ -86,7 +86,7 @@ def loss(blend_factor):
         return loss
 
     optimiser = Adam([blend_factor], lr=1e-1)
-    progress = tqdm(range(1000))
+    progress = range(1000)
     for i in progress:
         optimiser.zero_grad()
         loss_value = loss(blend_factor)