Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix: do not launch composition for atomized model #1324

Merged
merged 9 commits into from
Aug 26, 2024
4 changes: 3 additions & 1 deletion fedot/api/api_utils/predefined_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,9 @@ def _get_pipeline(self, use_input_preprocessing: bool = True) -> Pipeline:
else:
raise ValueError(f'{type(self.predefined_model)} is not supported as Fedot model')

verify_pipeline(pipelines, task_type=self.data.task.task_type, raise_on_failure=True)
# ISSUE #1317 Quick Fix: Skip pipeline verification for Atomized Model
Lopa10ko marked this conversation as resolved.
Show resolved Hide resolved
Lopa10ko marked this conversation as resolved.
Show resolved Hide resolved
if "atomized" not in pipelines.descriptive_id:
verify_pipeline(pipelines, task_type=self.data.task.task_type, raise_on_failure=True)

return pipelines

Expand Down
5 changes: 5 additions & 0 deletions fedot/api/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -166,6 +166,11 @@ def fit(self,
with fedot_composer_timer.launch_preprocessing():
self.train_data = self.data_processor.fit_transform(self.train_data)

# ISSUE #1317 Quick Fix: Do not launch composition for Atomized Model
Lopa10ko marked this conversation as resolved.
Show resolved Hide resolved
if init_asm := self.params.data.get('initial_assumption'):
Lopa10ko marked this conversation as resolved.
Show resolved Hide resolved
if predefined_model is None and "atomized" in init_asm.descriptive_id:
predefined_model = init_asm

with fedot_composer_timer.launch_fitting():
if predefined_model is not None:
# Fit predefined model and return it without composing
Expand Down
Loading