Fix dependencies

Signed-off-by: Beat Buesser <[email protected]>

Author: beat-buesser

tests/attacks/evasion/test_overload_attack.py

@@ -46,9 +46,10 @@ def test_generate(art_warning):
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])
+        # from ultralytics.nn.modules import Conv
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])
 
         model = YOLO("yolov5su.pt")
         py_model = PyTorchYolo(model=model, input_shape=(3, 640, 640), channels_first=True)
@@ -98,9 +99,9 @@ def test_check_params(art_warning):
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])
 
         model = YOLO("yolov5su.pt")
         py_model = PyTorchYolo(model=model, input_shape=(3, 640, 640), channels_first=True)

tests/attacks/evasion/test_steal_now_attack_later.py

@@ -49,7 +49,7 @@ def test_generate(art_warning):
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
         torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
         torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])
 
@@ -229,7 +229,7 @@ def test_check_params(art_warning):
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.C3])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Bottleneck])
         torch.serialization.add_safe_globals([ultralytics.nn.tasks.Detect])
-        torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
+        # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
         torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
         torch.serialization.add_safe_globals([ultralytics.nn.modules.DFL])