minor bug fixes

Author: nikhilwoodruff

.github/workflows/any_changes.yaml

@@ -20,6 +20,10 @@ jobs:
           uses: actions/setup-python@v2
           with:
             python-version: '3.11'
+        - uses: "google-github-actions/auth@v2"
+          with:
+            workload_identity_provider: "projects/322898545428/locations/global/workloadIdentityPools/policyengine-research-id-pool/providers/prod-github-provider"
+            service_account: "policyengine-research@policyengine-research.iam.gserviceaccount.com"
 
         - name: Install package
           run: uv pip install .[dev] --system

docs/concepts/simulation.ipynb

@@ -10,7 +10,6 @@
 from policyengine.outputs.macro.single.calculate_single_economy import (
     SingleEconomy,
 )
-from policyengine.utils.packages import get_country_package_version
 from typing import List, Dict
 
 
@@ -784,7 +783,8 @@ class CliffImpact(BaseModel):
 
 
 class EconomyComparison(BaseModel):
-    country_package_version: str
+    model_version: str
+    data_version: str
     budget: BudgetaryImpact
     detailed_budget: DetailedBudgetaryImpact
     decile: DecileImpact
@@ -847,7 +847,8 @@ def calculate_economy_comparison(
         cliff_impact = None
 
     return EconomyComparison(
-        country_package_version=get_country_package_version(country_id),
+        model_version=simulation.model_version,
+        data_version=simulation.data_version,
         budget=budgetary_impact_data,
         detailed_budget=detailed_budgetary_impact_data,
         decile=decile_impact_data,