Fix one more incompatibility with error messages

allenai · May 29, 2024 · 7a345ca · 7a345ca
1 parent cdb8b4a
commit 7a345ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tango/integrations/wandb/util.py b/tango/integrations/wandb/util.py
@@ -13,8 +13,10 @@ def is_missing_artifact_error(err: WandbError):
     Check if a specific W&B error is caused by a 404 on the artifact we're looking for.
     """
     # This is brittle, but at least we have a test for it.
-    return ("does not contain artifact" in err.message) or (
-        err.message == "'NoneType' object has no attribute 'get'"
+    return (
+        ("does not contain artifact" in err.message)
+        or ("Unable to fetch artifact with name" in err.message)
+        or (err.message == "'NoneType' object has no attribute 'get'")
     )  # This is a workaround for a bug in the wandb API