Moved 401 error into a separate block

databricks · Jun 17, 2024 · 594a85a · 594a85a
1 parent 1203a62
commit 594a85a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/databricks/sql/auth/retry.py b/src/databricks/sql/auth/retry.py
@@ -340,7 +340,13 @@ def should_retry(self, method: str, status_code: int) -> Tuple[bool, str]:
             return False, "200 codes are not retried"
 
         # Invalid Credentials error. Don't retry
-        if status_code == 403 or status_code == 401:
+        if status_code == 401:
+            raise NonRecoverableNetworkError(
+                "Received 401 - FORBIDDEN. Invalid authentication credentials."
+            )
+
+        # Invalid Credentials error. Don't retry
+        if status_code == 403:
             raise NonRecoverableNetworkError(
                 "Received 403 - FORBIDDEN. Confirm your authentication credentials."
             )