try

huggingface · Feb 7, 2024 · 9e5d347 · 9e5d347
1 parent 14f24b9
commit 9e5d347
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/models/llama/test_train_llama.py b/tests/models/llama/test_train_llama.py
@@ -1,9 +1,9 @@
 import unittest
 
-from ..test_model import TestModelTraining
+from ..test_model import ModelTrainingTestMixin
 
 
-class LLamaModelTrainingTest(TestModelTraining, unittest.TestCase):
+class LLamaModelTrainingTest(ModelTrainingTestMixin, unittest.TestCase):
 
     def prepare_command(self):
         tmp_dir = "my_dir"

diff --git a/tests/models/test_model.py b/tests/models/test_model.py
@@ -49,7 +49,7 @@
     import run_wav2vec2_pretraining_no_trainer
 
 
-class TestModelTraining:
+class ModelTrainingTestMixin:
 
     def prepare_command(self):
         raise NotImplementedError