ajoute start_computation_date

leximpact_survey_scenario/leximpact_survey_scenario.py

@@ -91,6 +91,7 @@ class LeximpactErfsSurveyScenario(AbstractSurveyScenario):
         computed_variables_to_dump=[],
         run_custom_initialize=True,
         monthly_incomes=False,
+        start_computation_date="2022-01",  # 1e mois des données qu'on a en input
     ):
         """
         Create a `LeximpactErfsSurveyScenario`
@@ -115,6 +116,7 @@ class LeximpactErfsSurveyScenario(AbstractSurveyScenario):
         self.period = period
         self.computed_variables_to_dump = computed_variables_to_dump
         self.run_custom_initialize = run_custom_initialize
+        self.start_computation_date = start_computation_date
 
         if dump or restore:
             assert dump_directory is not None
@@ -226,6 +228,10 @@ class LeximpactErfsSurveyScenario(AbstractSurveyScenario):
                 dump=dump,
                 memory_config=memory_config,
             )
+            for key in self.simulations.keys():
+                self.simulations[
+                    key
+                ].start_computation_date = self.start_computation_date
 
     def calibrate(
         self,
@@ -347,6 +353,9 @@ class LeximpactErfsSurveyScenario(AbstractSurveyScenario):
             simulation.weight_variable_by_entity = self.weight_variable_by_entity
             simulation.id_variable_by_entity_key = self.id_variable_by_entity_key
             self.simulations[prefix] = simulation
+            self.simulations[
+                prefix
+            ].start_computation_date = self.start_computation_date
         else:
             period = periods.period(self.annee_donnees)
             simulation = self.new_simulation(
@@ -356,6 +365,7 @@ class LeximpactErfsSurveyScenario(AbstractSurveyScenario):
                 data=data,
                 memory_config=memory_config,
             )
+            simulation.start_computation_date = self.start_computation_date
             if dump and prefix != "amendement":
                 log.info(f"Dumping {prefix} simulation")