Skip to content

Commit c415d40

Browse files
author
Beat Buesser
committed
Fix LGTM alerts
Signed-off-by: Beat Buesser <[email protected]>
1 parent e435ec6 commit c415d40

File tree

1 file changed

+5
-1
lines changed
  • art/estimators/certification/randomized_smoothing

1 file changed

+5
-1
lines changed

art/estimators/certification/randomized_smoothing/numpy.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from art.estimators.certification.randomized_smoothing.randomized_smoothing import RandomizedSmoothingMixin
2929
from art.estimators.classification import ClassifierMixin, ClassGradientsMixin
3030
import logging
31-
from typing import List, Union, TYPE_CHECKING
31+
from typing import List, Union, TYPE_CHECKING, Tuple
3232

3333
if TYPE_CHECKING:
3434
from art.utils import CLASSIFIER_LOSS_GRADIENTS_TYPE
@@ -69,6 +69,10 @@ def __init__(
6969
)
7070
self.classifier = classifier
7171

72+
@property
73+
def input_shape(self) -> Tuple[int, ...]:
74+
return self.classifier.input_shape
75+
7276
def _predict_classifier(self, x: np.ndarray, batch_size: int) -> np.ndarray:
7377
"""
7478
Perform prediction for a batch of inputs.

0 commit comments

Comments
 (0)