replace space to _ for name of test case

openvinotoolkit · Jul 15, 2024 · 54c8b2d · 54c8b2d
1 parent f83860b
commit 54c8b2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/torch/pruning/experimental/test_nodes_grouping.py b/tests/torch/pruning/experimental/test_nodes_grouping.py
@@ -488,14 +488,14 @@ def test_groups(desc: GroupTestDesc, mocker, tmp_path):
 BIG_MODEL_DESCS = [
     GroupTestDesc(
         model_desc=GeneralModelDesc(
-            model_name="MobileBERT big",
+            model_name="MobileBERT_big",
             input_info=[dict(sample_size=[1, 128], type="long")] * 4,
             model_builder=get_mobile_bert_big_model,
         ),
     ),
     GroupTestDesc(
         model_desc=GeneralModelDesc(
-            model_name="Swin big",
+            model_name="Swin_big",
             input_info=dict(sample_size=[1, 3, 224, 224]),
             model_builder=get_swin_tiny_model,
         )