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" + ] + } +} 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. 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()