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

…-auth credentials in auth interceptor (flyteorg#2591) Signed-off-by: Fabio Grätz <[email protected]> Co-authored-by: Fabio Grätz <[email protected]> Signed-off-by: mao3267 <[email protected]>
mao3267 · Aug 2, 2024 · 73a90e4 · 73a90e4
1 parent 0e1dae1
commit 73a90e4
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)