Skip to content

Commit 2999bf0

Browse files
author
Beat Buesser
committed
Updates for BaseEstimator.set_params
Signed-off-by: Beat Buesser <[email protected]>
1 parent 5319d95 commit 2999bf0

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

art/estimators/pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def _apply_preprocessing_gradient(self, x, gradients, fit=False):
264264

265265
elif len(self.preprocessing_operations) == 1 or (
266266
len(self.preprocessing_operations) == 2
267-
and isinstance(self.preprocessing[-1], (StandardisationMeanStd, StandardisationMeanStdPyTorch))
267+
and isinstance(self.preprocessing_operations[-1], (StandardisationMeanStd, StandardisationMeanStdPyTorch))
268268
):
269269
# Compatible with non-PyTorch defences if no chaining.
270270
for preprocess in self.preprocessing_operations[::-1]:

art/estimators/tensorflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -304,7 +304,7 @@ def _apply_preprocessing_gradient(self, x, gradients, fit=False):
304304

305305
elif len(self.preprocessing_operations) == 1 or (
306306
len(self.preprocessing_operations) == 2
307-
and isinstance(self.preprocessing[-1], (StandardisationMeanStd, StandardisationMeanStdTensorFlowV2))
307+
and isinstance(self.preprocessing_operations[-1], (StandardisationMeanStd, StandardisationMeanStdTensorFlowV2))
308308
):
309309
# Compatible with non-TensorFlow defences if no chaining.
310310
for preprocess in self.preprocessing_operations[::-1]:

0 commit comments

Comments
 (0)