bugfix: Catch - Can't load ... - error message

Lightning-AI · Jul 12, 2022 · 8396b1b · 8396b1b
1 parent 2d555ba
commit 8396b1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unittests/text/helpers.py b/tests/unittests/text/helpers.py
@@ -440,7 +440,7 @@ def skip_on_connection_issues(reason: str = "Unable to load checkpoints from Hug
 
     The tests run normally if no connection issue arises, and they're marked as skipped otherwise.
     """
-    _error_msg_starts = ["We couldn't connect to", "Connection error"]
+    _error_msg_starts = ["We couldn't connect to", "Connection error", "Can't load"]
 
     def test_decorator(function: Callable, *args: Any, **kwargs: Any) -> Optional[Callable]:
         @wraps(function)