nit

Signed-off-by: byhsu <byhsu@linkedin.com>
flyteorg · May 11, 2023 · 0d4684f · 0d4684f
1 parent 4615d2e
commit 0d4684f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flytekit/clients/auth/token_client.py b/flytekit/clients/auth/token_client.py
@@ -6,7 +6,7 @@
 import urllib.parse
 from dataclasses import dataclass
 from datetime import datetime, timedelta
-
+from flytekit import logger
 import requests
 
 from flytekit.clients.auth.exceptions import AuthenticationError, AuthenticationPending
@@ -150,7 +150,7 @@ def poll_token_endpoint(resp: DeviceCodeResponse, token_endpoint: str, client_id
         except AuthenticationPending:
             ...
         except Exception as e:
-            print("Authentication attempt failed: ", e)
+            logger.error("Authentication attempt failed: ", e)
             raise e
         print("Authentication Pending...")
         time.sleep(interval.total_seconds())