Check for multiple exception types #859

neuml · Jan 27, 2025 · 986da47 · 986da47
1 parent 5887f5f
commit 986da47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/python/testvectors/testwordvectors.py b/test/python/testvectors/testwordvectors.py
@@ -10,6 +10,7 @@
 
 import numpy as np
 
+from huggingface_hub.errors import HFValidationError
 from txtai.vectors import VectorsFactory
 
 
@@ -135,7 +136,7 @@ def testNoExist(self):
         """
 
         # Test non-existent path raises an exception
-        with self.assertRaises(IOError):
+        with self.assertRaises((IOError, HFValidationError)):
             VectorsFactory.create({"method": "words", "path": os.path.join(tempfile.gettempdir(), "noexist")}, None)
 
     def testTransform(self):