diff --git a/forge/test/models/pytorch/vision/autoencoder/results/reconstructed_image_linear.png b/forge/test/models/pytorch/vision/autoencoder/results/reconstructed_image_linear.png new file mode 100644 index 0000000000..3a6c0ac560 Binary files /dev/null and b/forge/test/models/pytorch/vision/autoencoder/results/reconstructed_image_linear.png differ diff --git a/forge/test/models/pytorch/vision/mobilenet/test_mobilenet_v1.py b/forge/test/models/pytorch/vision/mobilenet/test_mobilenet_v1.py index 519b7efd3e..7b0d0870ec 100644 --- a/forge/test/models/pytorch/vision/mobilenet/test_mobilenet_v1.py +++ b/forge/test/models/pytorch/vision/mobilenet/test_mobilenet_v1.py @@ -57,7 +57,12 @@ def test_mobilenetv1_basic(variant): loader.print_cls_results(co_out) -variants = [ModelVariant.MOBILENET_V1_075_192_HF, ModelVariant.MOBILENET_V1_100_224_HF] +variants = [ + ModelVariant.MOBILENET_V1_075_192_HF, + pytest.param( + ModelVariant.MOBILENET_V1_100_224_HF, marks=pytest.mark.my_check + ), + ] @pytest.mark.nightly diff --git a/forge/test/models/pytorch/vision/segformer/test_segformer.py b/forge/test/models/pytorch/vision/segformer/test_segformer.py index 0942a55bbb..05d6bd167e 100644 --- a/forge/test/models/pytorch/vision/segformer/test_segformer.py +++ b/forge/test/models/pytorch/vision/segformer/test_segformer.py @@ -29,7 +29,9 @@ variants_img_classification = [ ModelVariant.MIT_B0, ModelVariant.MIT_B1, - ModelVariant.MIT_B2, + pytest.param( + ModelVariant.MIT_B2, marks=pytest.mark.my_check + ), ModelVariant.MIT_B3, ModelVariant.MIT_B4, ModelVariant.MIT_B5, diff --git a/forge/test/models/pytorch/vision/yolo/test_yolo_v10.py b/forge/test/models/pytorch/vision/yolo/test_yolo_v10.py index d5abf4b388..dddbd65bf5 100644 --- a/forge/test/models/pytorch/vision/yolo/test_yolo_v10.py +++ b/forge/test/models/pytorch/vision/yolo/test_yolo_v10.py @@ -23,7 +23,9 @@ from test.models.pytorch.vision.yolo.model_utils.yolo_utils import YoloWrapper variants = [ - ModelVariant.YOLOV10X, + pytest.param( + ModelVariant.YOLOV10X, marks=pytest.mark.my_check + ), ModelVariant.YOLOV10N, ] diff --git a/pytest.ini b/pytest.ini index ecf354b216..912a3a5992 100644 --- a/pytest.ini +++ b/pytest.ini @@ -17,6 +17,7 @@ markers = run_in_pp: marks tests as run_in_pp # deprecated - tests that should run in push pipeline skip_model_analysis: marks tests as skip_model_analysis out_of_memory: marks tests as out_of_memory + my_check: testing specific cases # Where pytest should look for tests testpaths =