From cff5e6610db4dae2bed6c1aebc5f0a2fb645047f Mon Sep 17 00:00:00 2001
From: cgl <lallemand.chloe@gmail.com>
Date: Thu, 15 Feb 2024 14:40:11 +0100
Subject: [PATCH] Add config_files_directory option in survey_scenario
---
leximpact_prepare_data/pipeline_survey_scenario.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/leximpact_prepare_data/pipeline_survey_scenario.py b/leximpact_prepare_data/pipeline_survey_scenario.py
index 51c5aa3c..81c49d3b 100644
--- a/leximpact_prepare_data/pipeline_survey_scenario.py
+++ b/leximpact_prepare_data/pipeline_survey_scenario.py
@@ -9,6 +9,7 @@
from openfisca_core.taxbenefitsystems import TaxBenefitSystem
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
@@ -50,6 +51,7 @@ class PipelineErfsSurveyScenario(LeximpactErfsSurveyScenario):
def __init__(
self,
+ config_files_directory: str = default_config_files_directory,
annee_donnees: int = 2019,
period: int = 2023,
rebuild_input_data: bool = False,
@@ -80,6 +82,7 @@ def __init__(
)
super().__init__(
+ config_files_directory=config_files_directory,
annee_donnees=annee_donnees,
period=period,
rebuild_input_data=rebuild_input_data,
@@ -296,6 +299,7 @@ def save_current_survey(
period=year,
collection=collection,
survey_name=survey_name,
+ config_files_directory = self.data["config_files_directory"],
)
else:
data_frame_by_entity = self.simulations[
@@ -309,6 +313,7 @@ def save_current_survey(
period=period,
collection=collection,
survey_name=survey_name,
+ config_files_directory = self.data["config_files_directory"],
)
def custom_input_data_frame(self, input_data_frame, entity=None, **kwargs):
--
GitLab