Skip to content

Commit 4f8db1f

Browse files
authored
Merge pull request #1219 from Trusted-AI/development_issue_1217
Remove duplicate application of preprocessing in fallback BaseEstimator.fit_generator
2 parents 441d1a6 + 71f411f commit 4f8db1f

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

art/estimators/estimator.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -443,11 +443,8 @@ def fit_generator(self, generator: "DataGenerator", nb_epochs: int = 20, **kwarg
443443
): # type: ignore
444444
x, y = generator.get_batch()
445445

446-
# Apply preprocessing and defences
447-
x_preprocessed, y_preprocessed = self._apply_preprocessing(x, y, fit=True) # type: ignore
448-
449446
# Fit for current batch
450-
self.fit(x_preprocessed, y_preprocessed, nb_epochs=1, batch_size=generator.batch_size, **kwargs)
447+
self.fit(x, y, nb_epochs=1, batch_size=generator.batch_size, **kwargs)
451448

452449
@abstractmethod
453450
def get_activations(

0 commit comments

Comments
 (0)