Fix: Properly re-raise non-grpc exceptions during refreshing of proxy…

…-auth credentials in auth interceptor Signed-off-by: Fabio Grätz <[email protected]>
flyteorg · Jul 19, 2024 · 7370f6c · 7370f6c
1 parent a6a8651
commit 7370f6c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flytekit/clients/grpc_utils/auth_interceptor.py b/flytekit/clients/grpc_utils/auth_interceptor.py
@@ -61,6 +61,8 @@ def intercept_unary_unary(
         fut: grpc.Future = continuation(updated_call_details, request)
         e = fut.exception()
         if e:
+            if not hasattr(e, "code"):
+                raise e
             if e.code() == grpc.StatusCode.UNAUTHENTICATED or e.code() == grpc.StatusCode.UNKNOWN:
                 self._authenticator.refresh_credentials()
                 updated_call_details = self._call_details_with_auth_metadata(client_call_details)