diff --git a/forge/test/models/pytorch/vision/dla/test_dla.py b/forge/test/models/pytorch/vision/dla/test_dla.py index dd8095ca8..86a7d9ac9 100644 --- a/forge/test/models/pytorch/vision/dla/test_dla.py +++ b/forge/test/models/pytorch/vision/dla/test_dla.py @@ -10,7 +10,7 @@ from test.models.utils import Framework, Source, Task, build_module_name variants = [ - pytest.param("dla34", marks=[pytest.mark.push]), + "dla34", "dla46_c", "dla46x_c", "dla60",