diff --git a/src/databricks/sql/auth/authenticators.py b/src/databricks/sql/auth/authenticators.py index 8209931da..b5b1dfcb3 100644 --- a/src/databricks/sql/auth/authenticators.py +++ b/src/databricks/sql/auth/authenticators.py @@ -88,9 +88,11 @@ def _initial_get_token(self): ) self._access_token = access_token self._refresh_token = refresh_token - self._oauth_persistence.persist( - self._hostname, OAuthToken(access_token, refresh_token) - ) + + if self._oauth_persistence: + self._oauth_persistence.persist( + self._hostname, OAuthToken(access_token, refresh_token) + ) except Exception as e: logging.error(f"unexpected error in oauth initialization", e, exc_info=True) raise e