Fix 2

aimclub · Jul 2, 2024 · 930976e · 930976e
1 parent df022e3
commit 930976e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/fedot/api/time.py b/fedot/api/time.py
@@ -75,7 +75,9 @@ def launch_assumption_fit(self, n_folds: int):
         starting_time_for_assumption_fit = datetime.datetime.now()
         yield
         self.assumption_fit_spend_time_single_fold = \
-            (datetime.datetime.now() - starting_time_for_assumption_fit) * n_folds
+            (datetime.datetime.now() - starting_time_for_assumption_fit)
+        if n_folds is None:
+            n_folds = 1
         self.assumption_fit_spend_time = self.assumption_fit_spend_time_single_fold * n_folds
 
     def determine_resources_for_tuning(self):

diff --git a/test/unit/optimizer/test_external.py b/test/unit/optimizer/test_external.py
@@ -49,7 +49,7 @@ def test_external_static_optimizer(data_fixture, request):
                    preset='fast_train',
                    with_tuning=False,
                    optimizer=partial(StaticOptimizer, node_name='logit'),
-                   pop_size=2)
+                   pop_size=2, cv_folds=None)
     obtained_pipeline = automl.fit(train_data)
     automl.predict(test_data)