Skip to content
Snippets Groups Projects
Commit cff5e661 authored by Chloé Lallemand's avatar Chloé Lallemand
Browse files

Add config_files_directory option in survey_scenario

parent 7d418f22
No related branches found
No related tags found
1 merge request!120Resolve "Nettoyage config pipeline"
Pipeline #13005 failed
...@@ -9,6 +9,7 @@ ...@@ -9,6 +9,7 @@
from openfisca_core.taxbenefitsystems import TaxBenefitSystem from openfisca_core.taxbenefitsystems import TaxBenefitSystem
from openfisca_survey_manager.input_dataframe_generator import set_table_in_survey from openfisca_survey_manager.input_dataframe_generator import set_table_in_survey
from openfisca_survey_manager import default_config_files_directory
from leximpact_prepare_data.scenario_tools.helpers_survey_scenario import get_copules from leximpact_prepare_data.scenario_tools.helpers_survey_scenario import get_copules
...@@ -50,6 +51,7 @@ class PipelineErfsSurveyScenario(LeximpactErfsSurveyScenario): ...@@ -50,6 +51,7 @@ class PipelineErfsSurveyScenario(LeximpactErfsSurveyScenario):
def __init__( def __init__(
self, self,
config_files_directory: str = default_config_files_directory,
annee_donnees: int = 2019, annee_donnees: int = 2019,
period: int = 2023, period: int = 2023,
rebuild_input_data: bool = False, rebuild_input_data: bool = False,
...@@ -80,6 +82,7 @@ def __init__( ...@@ -80,6 +82,7 @@ def __init__(
) )
super().__init__( super().__init__(
config_files_directory=config_files_directory,
annee_donnees=annee_donnees, annee_donnees=annee_donnees,
period=period, period=period,
rebuild_input_data=rebuild_input_data, rebuild_input_data=rebuild_input_data,
...@@ -296,6 +299,7 @@ def save_current_survey( ...@@ -296,6 +299,7 @@ def save_current_survey(
period=year, period=year,
collection=collection, collection=collection,
survey_name=survey_name, survey_name=survey_name,
config_files_directory = self.data["config_files_directory"],
) )
else: else:
data_frame_by_entity = self.simulations[ data_frame_by_entity = self.simulations[
...@@ -309,6 +313,7 @@ def save_current_survey( ...@@ -309,6 +313,7 @@ def save_current_survey(
period=period, period=period,
collection=collection, collection=collection,
survey_name=survey_name, survey_name=survey_name,
config_files_directory = self.data["config_files_directory"],
) )
def custom_input_data_frame(self, input_data_frame, entity=None, **kwargs): def custom_input_data_frame(self, input_data_frame, entity=None, **kwargs):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment