Skip to content

Commit 5d9aff2

Browse files
author
Beat Buesser
committed
Fix type bug
Signed-off-by: Beat Buesser <[email protected]>
1 parent b52b94d commit 5d9aff2

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

art/estimators/estimator.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,17 +72,19 @@ def __init__(
7272
used for data preprocessing. The first value will be subtracted from the input and the results will be
7373
divided by the second value.
7474
"""
75+
from art.defences.postprocessor.postprocessor import Postprocessor
76+
from art.defences.preprocessor.preprocessor import Preprocessor
7577
self._model = model
7678
self._clip_values = clip_values
7779

7880
self.preprocessing_defences: Optional[List["Preprocessor"]]
79-
if isinstance(preprocessing_defences, "Preprocessor"):
81+
if isinstance(preprocessing_defences, Preprocessor):
8082
self.preprocessing_defences = [preprocessing_defences]
8183
else:
8284
self.preprocessing_defences = preprocessing_defences
8385

8486
self.postprocessing_defences: Optional[List["Postprocessor"]]
85-
if isinstance(postprocessing_defences, "Postprocessor"):
87+
if isinstance(postprocessing_defences, Postprocessor):
8688
self.postprocessing_defences = [postprocessing_defences]
8789
else:
8890
self.postprocessing_defences = postprocessing_defences

0 commit comments

Comments
 (0)