format

huggingface · Jul 3, 2024 · 4ee9c99 · 4ee9c99
1 parent 5e9be54
commit 4ee9c99
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -30,7 +30,7 @@
     "torch>=1.11",
     "transformers>=4.36.0,<4.43.0",
     "optimum @ git+https://github.com/huggingface/optimum.git",
-    #"optimum>=1.21.2,<1.22.0",
+    # "optimum>=1.21.2,<1.22.0",
     "datasets>=1.4.0",
     "sentencepiece",
     "setuptools",

diff --git a/tests/openvino/test_training.py b/tests/openvino/test_training.py
@@ -45,6 +45,7 @@
 from transformers.testing_utils import slow
 from transformers.trainer_utils import EvalPrediction, TrainOutput
 from transformers.utils import WEIGHTS_NAME
+from utils_tests import MODEL_NAMES
 
 from optimum.intel.openvino import OVTrainingArguments
 from optimum.intel.openvino.configuration import OVConfig
@@ -58,7 +59,6 @@
 from optimum.intel.openvino.utils import OV_XML_FILE_NAME
 from optimum.intel.utils.import_utils import is_transformers_version
 
-from utils_tests import MODEL_NAMES
 
 F32_CONFIG = {"INFERENCE_PRECISION_HINT": "f32"}
 
@@ -743,7 +743,7 @@ def check_ovmodel_reshaping(self, ovmodel: OVModel):
         compression_metrics=["compression_loss"],
     ),
     "structured_movement_sparsity": OVTrainerTestDescriptor(
-        model_id=MODEL_NAMES["wav2vec2-hf"]
+        model_id=MODEL_NAMES["wav2vec2-hf"],
         nncf_compression_config=[STRUCTURED_MOVEMENT_SPARSITY_CONFIG_FOR_WAV2VEC2],
         expected_binary_masks=48,
         compression_metrics=["compression_loss"],