fix(tests): allow model to be 1.0 or 1.1

EveryVoiceTTS · Jan 14, 2025 · 8a77971 · 8a77971
1 parent 15509bf
commit 8a77971
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/everyvoice/tests/test_model.py b/everyvoice/tests/test_model.py
@@ -404,7 +404,7 @@ def test_missing_model_version(self):
                     torch.save(m, ckpt_fn)
                     with mute_logger("everyvoice.config.text_config"):
                         model = ModelType.load_from_checkpoint(ckpt_fn)
-                    self.assertEqual(model._VERSION, "1.0")
+                    self.assertIn(model._VERSION, ["1.0", "1.1"])
 
     def test_newer_model_version(self):
         """