refactor: add log messages and TODO comments

aimclub · Aug 26, 2024 · d6555da · d6555da
1 parent 8d6a1fc
commit d6555da
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/fedot/api/api_utils/predefined_model.py b/fedot/api/api_utils/predefined_model.py
@@ -31,7 +31,10 @@ def _get_pipeline(self, use_input_preprocessing: bool = True) -> Pipeline:
         else:
             raise ValueError(f'{type(self.predefined_model)} is not supported as Fedot model')
 
-        if "atomized" not in pipelines.descriptive_id:
+        # TODO: Workaround for AtomizedModel
+        if "atomized" in pipelines.descriptive_id:
+            self.log.message("Pipeline verification for AtomizedModel currently unavailable")
+        else:
             verify_pipeline(pipelines, task_type=self.data.task.task_type, raise_on_failure=True)
 
         return pipelines

diff --git a/fedot/api/main.py b/fedot/api/main.py
@@ -166,9 +166,11 @@ def fit(self,
             with fedot_composer_timer.launch_preprocessing():
                 self.train_data = self.data_processor.fit_transform(self.train_data)
 
+        # TODO: Workaround for AtomizedModel
         init_asm = self.params.data.get('initial_assumption')
         if predefined_model is None:
             if isinstance(init_asm, Pipeline) and ("atomized" in init_asm.descriptive_id):
+                self.log.message('Composition for AtomizedModel currently unavailable')
                 predefined_model = init_asm
 
         with fedot_composer_timer.launch_fitting():