Finalize integration of BEYOND detector

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Dec 13, 2024 · bb73fd5 · bb73fd5
1 parent 4a40dcb
commit bb73fd5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/art/defences/detector/evasion/beyond_detector.py b/art/defences/detector/evasion/beyond_detector.py
@@ -164,7 +164,7 @@ def fit(self, x: np.ndarray, y: np.ndarray, batch_size: int = 128, nb_epochs: in
         k_minus_one_metrics = clean_metrics[:, self.var_K - 1]
         self.threshold = np.percentile(k_minus_one_metrics, q=self.percentile)
 
-    def detect(self, x: np.ndarray, batch_size: int = 128, **kwargs) -> tuple[np.ndarray, np.ndarray]:
+    def detect(self, x: np.ndarray, batch_size: int = 128, **kwargs) -> tuple[np.ndarray, np.ndarray]:  # type: ignore
         """
         Detect whether given samples are adversarial
 

diff --git a/tests/defences/detector/evasion/test_beyond_detector.py b/tests/defences/detector/evasion/test_beyond_detector.py
@@ -145,7 +145,7 @@ def test_beyond_detector(art_warning, get_default_cifar10_subset):
             augmentations=img_augmentations,
             aug_num=50,
             alpha=0.8,
-            K=20,
+            var_K=20,
             percentile=5,
         )