From 7a345ca956dfb74f61d5dd0a1295099bf5e94b9b Mon Sep 17 00:00:00 2001 From: Dirk Groeneveld Date: Tue, 28 May 2024 18:18:44 -0700 Subject: [PATCH] Fix one more incompatibility with error messages --- tango/integrations/wandb/util.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tango/integrations/wandb/util.py b/tango/integrations/wandb/util.py index 25ce7139e..06c7ca45d 100644 --- 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