fix: add optional_columns to result

openml · Mar 25, 2024 · c400961 · c400961
1 parent a357726
commit c400961
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/frameworks/FEDOT/exec_ts.py b/frameworks/FEDOT/exec_ts.py
@@ -81,14 +81,19 @@ def run(dataset, config):
         predictions.append(prediction)
         models_count += fedot.current_pipeline.length
 
+    optional_columns = dict(
+        repeated_item_id=np.load(dataset.repeated_item_id),
+        repeated_abs_seasonal_error=np.load(dataset.repeated_abs_seasonal_error),
+    )
     save_artifacts(fedot, config)
     return result(output_file=config.output_predictions_file,
                   predictions=np.hstack(predictions),
                   truth=truth_only,
                   target_is_encoded=False,
                   models_count=models_count,
                   training_duration=training_duration,
-                  predict_duration=predict_duration)
+                  predict_duration=predict_duration,
+                  optional_columns=optional_columns)
 
 
 def get_fedot_metrics(config):