diff --git a/art/estimators/certification/randomized_smoothing/randomized_smoothing.py b/art/estimators/certification/randomized_smoothing/randomized_smoothing.py index 6ce339c6cf..75f3bbd8ff 100644 --- a/art/estimators/certification/randomized_smoothing/randomized_smoothing.py +++ b/art/estimators/certification/randomized_smoothing/randomized_smoothing.py @@ -100,7 +100,7 @@ def predict(self, x: np.ndarray, batch_size: int = 128, verbose: bool = False, * # get class counts counts_pred = self._prediction_counts(x_i, batch_size=batch_size) top = counts_pred.argsort()[::-1] - # conversion to int + # conversion to int count1 = int(np.max(counts_pred)) count2 = int(counts_pred[top[1]])