test: fix integration test

aimclub · Aug 26, 2024 · b7466e0 · b7466e0
1 parent f3fc0ca
commit b7466e0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fedot/api/main.py b/fedot/api/main.py
@@ -167,8 +167,9 @@ def fit(self,
                 self.train_data = self.data_processor.fit_transform(self.train_data)
 
         init_asm = self.params.data.get('initial_assumption')
-        if (predefined_model is None) and init_asm and ("atomized" in init_asm.descriptive_id):
-            predefined_model = init_asm
+        if (predefined_model is None):
+            if isinstance(init_asm, Pipeline) and ("atomized" in init_asm.descriptive_id):
+                predefined_model = init_asm
 
         with fedot_composer_timer.launch_fitting():
             if predefined_model is not None: