Skip to content

Commit c58070d

Browse files
author
Beat Buesser
committed
Update type check
Signed-off-by: Beat Buesser <[email protected]>
1 parent 5d9aff2 commit c58070d

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

art/estimators/estimator.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ def __init__(
7474
"""
7575
from art.defences.postprocessor.postprocessor import Postprocessor
7676
from art.defences.preprocessor.preprocessor import Preprocessor
77+
7778
self._model = model
7879
self._clip_values = clip_values
7980

@@ -120,6 +121,9 @@ def get_params(self) -> Dict[str, Any]:
120121
return params
121122

122123
def _check_params(self) -> None:
124+
from art.defences.postprocessor.postprocessor import Postprocessor
125+
from art.defences.preprocessor.preprocessor import Preprocessor
126+
123127
if self._clip_values is not None:
124128
if len(self._clip_values) != 2:
125129
raise ValueError(
@@ -135,7 +139,7 @@ def _check_params(self) -> None:
135139

136140
if isinstance(self.preprocessing_defences, list):
137141
for preproc_defence in self.preprocessing_defences:
138-
if not isinstance(preproc_defence, "Preprocessor"):
142+
if not isinstance(preproc_defence, Preprocessor):
139143
raise ValueError(
140144
"All preprocessing defences have to be instance of "
141145
"art.defences.preprocessor.preprocessor.Preprocessor."
@@ -149,7 +153,7 @@ def _check_params(self) -> None:
149153
)
150154
if isinstance(self.postprocessing_defences, list):
151155
for postproc_defence in self.postprocessing_defences:
152-
if not isinstance(postproc_defence, "Postprocessor"):
156+
if not isinstance(postproc_defence, Postprocessor):
153157
raise ValueError(
154158
"All postprocessing defences have to be instance of "
155159
"art.defences.postprocessor.postprocessor.Postprocessor."

0 commit comments

Comments
 (0)