Skip to content

Commit e42ab42

Browse files
author
Beat Buesser
committed
Mypy updates
Signed-off-by: Beat Buesser <[email protected]>
1 parent aa881b5 commit e42ab42

File tree

4 files changed

+5
-12
lines changed

4 files changed

+5
-12
lines changed

art/estimators/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def _update_preprocessing_operations(self):
109109
raise ValueError("Preprocessing argument not recognised.")
110110

111111
@staticmethod
112-
def _set_preprocessing(preprocessing: Union["PREPROCESSING_TYPE", "Preprocessor"]) -> "Preprocessor":
112+
def _set_preprocessing(preprocessing: Optional[Union["PREPROCESSING_TYPE", "Preprocessor"]]) -> Optional["Preprocessor"]:
113113
from art.defences.preprocessor.preprocessor import Preprocessor
114114

115115
if preprocessing is None:

art/estimators/generation/tensorflow.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -201,15 +201,6 @@ def model(self) -> "tf.Tensor":
201201
"""
202202
return self._model
203203

204-
@property
205-
def input_ph(self) -> "tf.Placeholder":
206-
"""
207-
Returns the encoding seed input of the generator of shape `(batch_size, encoding_length)`.
208-
209-
:return: The encoding seed input of the generator.
210-
"""
211-
return self._input_ph
212-
213204
@property
214205
def encoding_length(self) -> int:
215206
"""

art/metrics/privacy/membership_leakage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
from art.utils import check_and_transform_label_format, is_probability
2828

2929
if TYPE_CHECKING:
30-
from art.estimators.classification import Classifier
30+
from art.estimators.classification.classifier import Classifier
3131

3232

3333
def PDTP(

art/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,9 @@
5757
from art.defences.preprocessor.preprocessor import Preprocessor
5858

5959
PREPROCESSING_TYPE = Optional[
60-
Tuple[Union[int, float, np.ndarray], Union[int, float, np.ndarray]], Preprocessor, Tuple[Preprocessor, ...]
60+
Union[
61+
Tuple[Union[int, float, np.ndarray], Union[int, float, np.ndarray]], Preprocessor, Tuple[Preprocessor, ...]
62+
]
6163
]
6264

6365
from art.estimators.classification.classifier import (

0 commit comments

Comments
 (0)