mirror of
https://github.com/neuml/txtai.git
synced 2026-10-11 21:39:55 +00:00
test: cover the tqdm library stub
This commit is contained in:
@@ -34,6 +34,8 @@ class TestLibrary(unittest.TestCase):
|
||||
"torch.nn",
|
||||
"torch.onnx",
|
||||
"torch.utils.data",
|
||||
"tqdm",
|
||||
"tqdm.auto",
|
||||
"yaml",
|
||||
]
|
||||
|
||||
@@ -95,6 +97,9 @@ class TestLibrary(unittest.TestCase):
|
||||
with self.assertRaises(ImportError):
|
||||
lib.torch().device
|
||||
|
||||
with self.assertRaises(ImportError):
|
||||
lib.tqdm().tqdm
|
||||
|
||||
with self.assertRaises(ImportError):
|
||||
lib.transformers().AutoModel
|
||||
|
||||
|
||||
Reference in New Issue
Block a user