From ea22b3e598d803e84cd7990a102a0341df5d136c Mon Sep 17 00:00:00 2001 From: Max Ghenis Date: Sat, 17 Jan 2026 16:33:55 -0500 Subject: [PATCH 1/3] Add policyengine-claude plugin auto-install Co-Authored-By: Claude --- .claude/settings.json | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 .claude/settings.json diff --git a/.claude/settings.json b/.claude/settings.json new file mode 100644 index 000000000..647162e29 --- /dev/null +++ b/.claude/settings.json @@ -0,0 +1,10 @@ +{ + "plugins": { + "marketplaces": [ + "PolicyEngine/policyengine-claude" + ], + "auto_install": [ + "country-models@policyengine-claude" + ] + } +} From b1264e552d50b18de40c77ce495ff82720904723 Mon Sep 17 00:00:00 2001 From: Max Ghenis Date: Sun, 18 Jan 2026 15:35:34 -0500 Subject: [PATCH 2/3] Add changelog entry --- changelog_entry.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/changelog_entry.yaml b/changelog_entry.yaml index e69de29bb..b46584867 100644 --- a/changelog_entry.yaml +++ b/changelog_entry.yaml @@ -0,0 +1,4 @@ +- bump: minor + changes: + changed: + - Added policyengine-claude plugin auto-install configuration. From 694c6d03f24e58e180175234f2aa6cb4b0dfb7c3 Mon Sep 17 00:00:00 2001 From: Max Ghenis Date: Sun, 18 Jan 2026 15:52:00 -0500 Subject: [PATCH 3/3] Fix black 26.1.0 formatting --- policyengine_api/data/congressional_districts.py | 1 - policyengine_api/libs/simulation_api.py | 1 - policyengine_api/routes/ai_prompt_routes.py | 1 - policyengine_api/routes/household_routes.py | 1 - policyengine_api/routes/report_output_routes.py | 1 - policyengine_api/routes/simulation_routes.py | 1 - tests/conftest.py | 1 - tests/fixtures/libs/simulation_api_modal.py | 1 - tests/fixtures/services/household_fixtures.py | 1 - tests/fixtures/services/tracer_analysis_service.py | 1 - tests/fixtures/services/user_service.py | 1 - tests/unit/services/test_tracer_analysis_service.py | 1 - 12 files changed, 12 deletions(-) diff --git a/policyengine_api/data/congressional_districts.py b/policyengine_api/data/congressional_districts.py index 7aa54ab8c..8c52c4e64 100644 --- a/policyengine_api/data/congressional_districts.py +++ b/policyengine_api/data/congressional_districts.py @@ -11,7 +11,6 @@ from pydantic import BaseModel, Field - # Mapping of state codes to full state names STATE_CODE_TO_NAME = { "AL": "Alabama", diff --git a/policyengine_api/libs/simulation_api.py b/policyengine_api/libs/simulation_api.py index 1fbd12b48..3d4554de9 100644 --- a/policyengine_api/libs/simulation_api.py +++ b/policyengine_api/libs/simulation_api.py @@ -8,7 +8,6 @@ from policyengine_api.gcp_logging import logger from google.cloud.workflows import executions_v1 - load_dotenv() ExecutionState = executions_v1.types.Execution.State diff --git a/policyengine_api/routes/ai_prompt_routes.py b/policyengine_api/routes/ai_prompt_routes.py index c377c51b5..a15bd16dd 100644 --- a/policyengine_api/routes/ai_prompt_routes.py +++ b/policyengine_api/routes/ai_prompt_routes.py @@ -8,7 +8,6 @@ from werkzeug.exceptions import NotFound, BadRequest import json - ai_prompt_bp = Blueprint("ai_prompt", __name__) ai_prompt_service = AIPromptService() diff --git a/policyengine_api/routes/household_routes.py b/policyengine_api/routes/household_routes.py index 893d6defd..0961d8cd3 100644 --- a/policyengine_api/routes/household_routes.py +++ b/policyengine_api/routes/household_routes.py @@ -8,7 +8,6 @@ validate_country, ) - household_bp = Blueprint("household", __name__) household_service = HouseholdService() diff --git a/policyengine_api/routes/report_output_routes.py b/policyengine_api/routes/report_output_routes.py index 4dfb9218a..b2c5502d1 100644 --- a/policyengine_api/routes/report_output_routes.py +++ b/policyengine_api/routes/report_output_routes.py @@ -6,7 +6,6 @@ from policyengine_api.constants import CURRENT_YEAR from policyengine_api.utils.payload_validators import validate_country - report_output_bp = Blueprint("report_output", __name__) report_output_service = ReportOutputService() diff --git a/policyengine_api/routes/simulation_routes.py b/policyengine_api/routes/simulation_routes.py index c1210d97d..86e6f0ddf 100644 --- a/policyengine_api/routes/simulation_routes.py +++ b/policyengine_api/routes/simulation_routes.py @@ -6,7 +6,6 @@ from policyengine_api.utils.payload_validators import validate_country from policyengine_api.constants import COUNTRY_PACKAGE_VERSIONS - simulation_bp = Blueprint("simulation", __name__) simulation_service = SimulationService() diff --git a/tests/conftest.py b/tests/conftest.py index 217f37342..f604176cd 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -8,7 +8,6 @@ import pytest from policyengine_api.api import app - # Add the project root directory to PYTHONPATH root_dir = Path(__file__).parent sys.path.append(str(root_dir)) diff --git a/tests/fixtures/libs/simulation_api_modal.py b/tests/fixtures/libs/simulation_api_modal.py index 216cf5bf1..64ce139e7 100644 --- a/tests/fixtures/libs/simulation_api_modal.py +++ b/tests/fixtures/libs/simulation_api_modal.py @@ -16,7 +16,6 @@ MODAL_EXECUTION_STATUS_FAILED, ) - # Mock data constants MOCK_MODAL_JOB_ID = "fc-abc123xyz" MOCK_MODAL_BASE_URL = "https://test-modal-api.modal.run" diff --git a/tests/fixtures/services/household_fixtures.py b/tests/fixtures/services/household_fixtures.py index f84d99c95..54d49291f 100644 --- a/tests/fixtures/services/household_fixtures.py +++ b/tests/fixtures/services/household_fixtures.py @@ -2,7 +2,6 @@ import json from unittest.mock import patch - valid_request_body = { "data": {"people": {"person1": {"age": 30, "income": 50000}}}, "label": "Test Household", diff --git a/tests/fixtures/services/tracer_analysis_service.py b/tests/fixtures/services/tracer_analysis_service.py index f137c206b..1a1262dc0 100644 --- a/tests/fixtures/services/tracer_analysis_service.py +++ b/tests/fixtures/services/tracer_analysis_service.py @@ -5,7 +5,6 @@ ) from unittest.mock import patch - valid_tracer_output = [ " snap<2027, (default)> = [6769.799]", " snap<2027-01, (default)> = [561.117]", diff --git a/tests/fixtures/services/user_service.py b/tests/fixtures/services/user_service.py index 2a77d2c9d..1ab78bc80 100644 --- a/tests/fixtures/services/user_service.py +++ b/tests/fixtures/services/user_service.py @@ -1,6 +1,5 @@ import pytest - valid_user_record = { "user_id": 1, "auth0_id": "123", diff --git a/tests/unit/services/test_tracer_analysis_service.py b/tests/unit/services/test_tracer_analysis_service.py index 1e87c41a6..eeb08c9b0 100644 --- a/tests/unit/services/test_tracer_analysis_service.py +++ b/tests/unit/services/test_tracer_analysis_service.py @@ -7,7 +7,6 @@ from tests.fixtures.services.tracer_analysis_service import * - logger = logging.getLogger(__name__) test_service = TracerAnalysisService()