fix: mobilenet_v2 model type

JingyaHuang · dacorvo · commit 88db919f2db8 · 2025-11-05T17:10:08.000+01:00
diff --git a/tests/inference/transformers/test_modeling.py b/tests/inference/transformers/test_modeling.py
@@ -1060,7 +1060,7 @@ class NeuronModelForSemanticSegmentationIntegrationTest(NeuronModelTestMixin):
     TASK = "semantic-segmentation"
     ATOL_FOR_VALIDATION = 1e-3
     SUPPORTED_ARCHITECTURES = [
-        "mobilenet-v2",
+        "mobilenet_v2",
         "mobilevit",
     ]
 

Original file line number	Diff line number	Diff line change
`@@ -1060,7 +1060,7 @@ class NeuronModelForSemanticSegmentationIntegrationTest(NeuronModelTestMixin):`
`1060`	`1060`	`TASK = "semantic-segmentation"`
`1061`	`1061`	`ATOL_FOR_VALIDATION = 1e-3`
`1062`	`1062`	`SUPPORTED_ARCHITECTURES = [`
`1063`		`- "mobilenet-v2",`
	`1063`	`+ "mobilenet_v2",`
`1064`	`1064`	`"mobilevit",`
`1065`	`1065`	`]`
`1066`	`1066`