Fix dependencies

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

Author: beat-buesser

tests/attacks/evasion/test_overload_attack.py

@@ -32,6 +32,7 @@ def test_generate(art_warning):
     try:
         import torch
         import ultralytics
+        from ultralytics.nn.modules import Conv
         from ultralytics import YOLO
 
         torch.serialization.add_safe_globals([torch.nn.modules.container.Sequential])
@@ -46,6 +47,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([Conv])
         # from ultralytics.nn.modules import Conv
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Conv])
         # torch.serialization.add_safe_globals([ultralytics.nn.modules.Concat])
@@ -86,6 +88,7 @@ def test_check_params(art_warning):
         import torch
         import ultralytics
         from ultralytics import YOLO
+        from ultralytics.nn.modules import Conv
 
         torch.serialization.add_safe_globals([torch.nn.modules.container.Sequential])
         torch.serialization.add_safe_globals([torch.nn.modules.container.ModuleList])
@@ -99,6 +102,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([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])