File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed
art/estimators/certification/randomized_smoothing Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change 2828from art .estimators .certification .randomized_smoothing .randomized_smoothing import RandomizedSmoothingMixin
2929from art .estimators .classification import ClassifierMixin , ClassGradientsMixin
3030import logging
31- from typing import List , Union , TYPE_CHECKING
31+ from typing import List , Union , TYPE_CHECKING , Tuple
3232
3333if 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.
You can’t perform that action at this time.
0 commit comments