Update tests: try_to_load_from_cache can return None

hfg-gmuend · May 27, 2024 · bcc385c · bcc385c
1 parent 8005503
commit bcc385c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/tests.py b/test/tests.py
@@ -84,6 +84,8 @@ def test_install_huggingface(self):
     def test_install_huggingface_cached(self):
         filepath = try_to_load_from_cache(
             repo_id="sd-concepts-library/moebius", filename="learned_embeds.bin")
+
+        self.assertTrue(isinstance(filepath, str))
         self.assertTrue(os.path.exists(filepath))
         self.assertIsInstance(filepath, str)
         self.assertFalse(filepath == _CACHED_NO_EXIST)