Skip to content

Commit

Permalink
fix unexpected inplace operations
Browse files Browse the repository at this point in the history
  • Loading branch information
valer1435 committed Aug 7, 2023
1 parent fab0a20 commit 685472a
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions fedot/core/operations/operation.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from abc import abstractmethod
from copy import deepcopy
from typing import Optional, Union, Dict, Any

from golem.core.log import default_log
Expand Down Expand Up @@ -80,6 +81,7 @@ def fit(self, params: Optional[Union[OperationParameters, dict]], data: InputDat
Returns:
tuple: trained operation and prediction on train data
"""
data = deepcopy(data)
self._init(data.task, params=params, n_samples_data=data.features.shape[0])

self.fitted_operation = self._eval_strategy.fit(train_data=data)
Expand Down Expand Up @@ -119,6 +121,7 @@ def predict_for_fit(self, fitted_operation, data: InputData, params: Optional[Op
def _predict(self, fitted_operation, data: InputData, params: Optional[OperationParameters] = None,
output_mode: str = 'default', is_fit_stage: bool = False):

data = deepcopy(data)
is_main_target = data.supplementary_data.is_main_target
data_flow_length = data.supplementary_data.data_flow_length
self._init(data.task, output_mode=output_mode, params=params, n_samples_data=data.features.shape[0])
Expand Down

0 comments on commit 685472a

Please sign in to comment.