fix: Log actual exception in case of NeptuneAuthTokenExpired (#1531)

neptune-ai · Oct 17, 2023 · 87a3c0d · 87a3c0d
1 parent 21ec3f6
commit 87a3c0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -7,6 +7,7 @@
 - Run async callback in new daemon thread ([#1521](https://github.com/neptune-ai/neptune-client/pull/1521))
 - Better handle bool values of `git_ref` param in `init_run` ([#1525](https://github.com/neptune-ai/neptune-client/pull/1525))
 - Updated management docstrings ([#1500](https://github.com/neptune-ai/neptune-client/pull/1500))
+- Fix error massage in case of NeptuneAuthTokenExpired ([#1531](https://github.com/neptune-ai/neptune-client/pull/1531))
 
 ### Changes
 - Safety (errors suppressing) execution mode ([#1503](https://github.com/neptune-ai/neptune-client/pull/1503))

diff --git a/src/neptune/common/backends/utils.py b/src/neptune/common/backends/utils.py
@@ -101,7 +101,8 @@ def wrapper(*args, **kwargs):
                 time.sleep(wait_time)
                 last_exception = e
                 continue
-            except NeptuneAuthTokenExpired:
+            except NeptuneAuthTokenExpired as e:
+                last_exception = e
                 continue
             except HTTPUnauthorized:
                 raise Unauthorized()