fix: add compute_config layer on output_folder and history dir setting

aimclub · Jan 17, 2025 · de8b672 · de8b672
1 parent 2b590f1
commit de8b672
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/benchmark/abstract_bench.py b/benchmark/abstract_bench.py
@@ -57,10 +57,8 @@ def run(self):
     def evaluate_loop(self, dataset, experiment_setup: dict = None):
         matplotlib.use('TkAgg')
         train_data, test_data = DataLoader(dataset_name=dataset).load_data()
-        experiment_setup['output_folder'] = experiment_setup['output_folder'] + \
-            f'/{dataset}'
-        experiment_setup['history_dir'] = './composition_results' + \
-            f'/{dataset}'
+        experiment_setup['compute_config']['output_folder'] += f'/{dataset}'
+        experiment_setup['compute_config']['history_dir'] = f'./composition_results/{dataset}'
         model = FedotIndustrial(**experiment_setup)
         model.fit(train_data)
         prediction = model.predict(test_data)