|
16 | 16 | "outputs": [], |
17 | 17 | "source": [ |
18 | 18 | "from policyengine_core.country_template import (\n", |
19 | | - " CountryTaxBenefitSystem, # Holds all policy-related data.\n", |
20 | | - " Simulation, # General purpose: enables users to simulate policy on new data.\n", |
21 | | - " Microsimulation, # Specific purpose: enables users to simulate policy on large datasets.\n", |
22 | 19 | " COUNTRY_DIR, # Path to the country directory.\n", |
23 | 20 | " DATASETS, # List of datasets available for this country (to use with `Microsimulation`).\n", |
| 21 | + " CountryTaxBenefitSystem, # Holds all policy-related data.\n", |
| 22 | + " Microsimulation, # Specific purpose: enables users to simulate policy on large datasets.\n", |
| 23 | + " Simulation, # General purpose: enables users to simulate policy on new data.\n", |
24 | 24 | ")" |
25 | 25 | ] |
26 | 26 | }, |
|
52 | 52 | "metadata": {}, |
53 | 53 | "outputs": [], |
54 | 54 | "source": [ |
55 | | - "import os\n", |
56 | | - "from typing import Dict, Type\n", |
57 | | - "from policyengine_core.data.dataset import Dataset\n", |
58 | | - "from policyengine_core.populations.population import Population\n", |
59 | | - "from policyengine_core.taxbenefitsystems import TaxBenefitSystem\n", |
60 | 55 | "from policyengine_core.country_template import entities\n", |
61 | | - "from policyengine_core.country_template.situation_examples import couple\n", |
62 | | - "from policyengine_core.simulations import Simulation as CoreSimulation\n", |
63 | | - "from policyengine_core.simulations import (\n", |
64 | | - " Microsimulation as CoreMicrosimulation,\n", |
65 | | - ")\n", |
| 56 | + "from policyengine_core.country_template.constants import COUNTRY_DIR\n", |
66 | 57 | "from policyengine_core.country_template.data.datasets.country_template_dataset import (\n", |
67 | 58 | " CountryTemplateDataset,\n", |
68 | 59 | ")\n", |
69 | | - "from pathlib import Path\n", |
70 | | - "import logging\n", |
71 | | - "from policyengine_core.country_template.constants import COUNTRY_DIR\n", |
| 60 | + "from policyengine_core.simulations import (\n", |
| 61 | + " Microsimulation as CoreMicrosimulation,\n", |
| 62 | + ")\n", |
| 63 | + "from policyengine_core.simulations import Simulation as CoreSimulation\n", |
| 64 | + "from policyengine_core.taxbenefitsystems import TaxBenefitSystem\n", |
72 | 65 | "\n", |
73 | 66 | "DATASETS = [CountryTemplateDataset] # Important: must be instantiated\n", |
74 | 67 | "\n", |
|
0 commit comments