Fix bug in generation script

Author: nikhilwoodruff

.github/workflows/push.yaml

@@ -45,20 +45,12 @@ jobs:
           POLICYENGINE_US_DATA_GITHUB_TOKEN: ${{ secrets.POLICYENGINE_US_DATA_GITHUB_TOKEN }}
       - name: Build datasets
         run: make data
+        env:
+          LITE_MODE: true
       - name: Run tests
         run: pytest
       - name: Test documentation builds
         run: make documentation
-      - name: Upload CPS 2024
-        uses: actions/upload-artifact@v4
-        with:
-          name: cps_2024.h5
-          path: policyengine_us_data/storage/cps_2024.h5
-      - name: Upload ECPS 2024
-        uses: actions/upload-artifact@v4
-        with:
-          name: enhanced_cps_2024.h5
-          path: policyengine_us_data/storage/enhanced_cps_2024.h5
       - name: Build Jupyter Book
         run: make documentation
       - name: Deploy documentation

policyengine_us_data/datasets/cps/extended_cps.py

@@ -118,7 +118,7 @@ def generate(self):
         print(
             f"Predicting imputed values took {time.time() - start:.2f} seconds"
         )
-
+        cps_sim = Microsimulation(dataset=self.cps)
         data = cps_sim.dataset.load_dataset()
         new_data = {}