From 54c8b2da7a43550ded067599b261c7405461c7cf Mon Sep 17 00:00:00 2001 From: Alexander Dokuchaev Date: Tue, 16 Jul 2024 01:39:58 +0300 Subject: [PATCH] replace space to _ for name of test case --- tests/torch/pruning/experimental/test_nodes_grouping.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/torch/pruning/experimental/test_nodes_grouping.py b/tests/torch/pruning/experimental/test_nodes_grouping.py index 931bc3e42a8..d906e1e8eda 100644 --- 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, )