diff --git a/tests/conftest.py b/tests/conftest.py index bdae295c..b54159ec 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -20,7 +20,6 @@ experiments_names = [ "CIFAR10_CNN", "CIFAR10_CNN_Adv", - "CIFAR10_RobustBench", "ImageNet_Timm", "COCO_TorchvisionFasterRCNN", "COCO_TorchvisionFasterRCNN_Adv", diff --git a/tests/test_experiments.py b/tests/test_experiments.py index e6f4f670..d28cf0fa 100644 --- a/tests/test_experiments.py +++ b/tests/test_experiments.py @@ -264,7 +264,7 @@ def test_resume(tmpdir): overrides_yaml.write( "\n".join( [ - "- experiment=CIFAR10_RobustBench", + "- experiment=CIFAR10_CNN", "- datamodule=dummy_classification", "- datamodule.ims_per_batch=2", "- datamodule.num_workers=0",