From 0b4bb68325301e62de35cd3774dad617adda90ee Mon Sep 17 00:00:00 2001 From: Beat Buesser Date: Mon, 16 Dec 2024 22:16:30 +0100 Subject: [PATCH] Apply formatting Signed-off-by: Beat Buesser --- .../certification/randomized_smoothing/randomized_smoothing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]])