Test microsimulation works without optional dependencies

.github/workflows/pr_code_changes.yaml

@@ -48,6 +48,9 @@ jobs:
       - name: Test specific core import
         run: python -c "from policyengine_core.data import Dataset; print('Core import OK')"
 
+      - name: Test Microsimulation with direct dependencies only
+        run: python -m pytest policyengine_us_data/tests/test_datasets/test_data_us_pipeline.py -v
+
   Test:
       runs-on: ubuntu-latest
       needs: Lint

changelog_entry.yaml

@@ -0,0 +1,4 @@
+- bump: minor
+  changes:
+    changed:
+    - Added test to ensure microsimulation runs with only us-data direct dependencies.

policyengine_us_data/datasets/cps/enhanced_cps.py

@@ -21,12 +21,6 @@
 from pathlib import Path
 
 
-try:
-    import torch
-except ImportError:
-    torch = None
-
-
 def reweight(
     original_weights,
     loss_matrix,
@@ -38,6 +32,8 @@ def reweight(
     init_mean=0.999,  # initial proportion with non-zero weights, set near 0
     temperature=0.5,  # Usual values .5 to 3, .5 was working better
 ):
+    import torch
+
     target_names = np.array(loss_matrix.columns)
     is_national = loss_matrix.columns.str.startswith("nation/")
     loss_matrix = torch.tensor(loss_matrix.values, dtype=torch.float32)

policyengine_us_data/tests/test_datasets/test_data_us_pipeline.py

@@ -0,0 +1,11 @@
+"""
+Tests to ensure Microsimulations can run with the us-data direct dependencies only.
+"""
+
+
+def test_microsimulation_runs():
+    from policyengine_us import Microsimulation
+    from policyengine_us_data.datasets.cps import EnhancedCPS_2024
+
+    sim = Microsimulation(dataset=EnhancedCPS_2024)
+    sim.calculate("employment_income", map_to="household", period=2025)