Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions swanlab/core_python/client/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,14 @@
from typing import Optional, Tuple, Dict, Union, List, AnyStr

import requests
from urllib3.exceptions import (
MaxRetryError,
TimeoutError,
NewConnectionError,
ConnectionError,
ReadTimeoutError,
ConnectTimeoutError,
)

from swanlab.error import NetworkError, ApiError
from swanlab.log import swanlog
Expand Down Expand Up @@ -445,6 +453,16 @@ def wrapper(*args, **kwargs) -> Tuple[Optional[Union[dict, str]], Optional[Excep
return None, NetworkError()
except requests.exceptions.ConnectionError:
return None, NetworkError()
# Catch urllib3 specific errors
except (
MaxRetryError,
TimeoutError,
NewConnectionError,
ConnectionError,
ReadTimeoutError,
ConnectTimeoutError,
):
return None, NetworkError()
except Exception as e:
return None, e

Expand Down