Skip to content
Merged
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -209,5 +209,6 @@ __marimo__/
# Data files
*.csv
*.h5
*.db

.DS_Store
4 changes: 4 additions & 0 deletions changelog_entry.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
- bump: minor
changes:
changed:
- Added logic to rescale calibration targets from a database.
2 changes: 2 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@ dependencies = [
"tables",
"policyengine-core>=3.6.4",
"microdf-python",
"sqlalchemy",
"huggingface_hub",
]

[project.optional-dependencies]
Expand Down
1 change: 1 addition & 0 deletions src/policyengine_data/calibration/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from .target_rescaling import download_database, rescale_calibration_targets
3 changes: 3 additions & 0 deletions src/policyengine_data/calibration/calibrate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
"""
This file will contain the logic for calibrating policy engine data from start to end. It will include functions for target rescaling, matrix creation, household duplication and assignment to new geographic areas, and final calibration.
"""
Loading