fix: Avoid NA in daily disaggregation

adevinta · May 21, 2024 · ad44525 · ad44525
1 parent ff4627c
commit ad44525
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/forecastout/core.py b/forecastout/core.py
@@ -157,7 +157,9 @@ def __get_daily_disaggregation_shares(self):
         )
         self.log.info(">> 3.3. Train disaggregation models")
         disaggregation_model_trainer = DisaggregationModelTrainer(
-            df_train=df_train_test.loc[df_train_test['test'] == 0].copy(),
+            df_train=(
+                df_train_test.loc[df_train_test['test'] == 0].dropna().copy()
+             ),
             config=self.config
         )
         disaggregation_model_trained = disaggregation_model_trainer.train(