Skip to content

Commit 106f33a

Browse files
committed
dont fail on init if fd unavailable
Signed-off-by: Praneeth Bedapudi <[email protected]>
1 parent 0092489 commit 106f33a

File tree

2 files changed

+17
-10
lines changed

2 files changed

+17
-10
lines changed

Diff for: clients/python/fdclient/client.py

+16-9
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,19 @@ def __init__(self, server_url, request_timeout, compression=True):
2626
self.local_storage = threading.local()
2727
self.requests_session = requests.Session()
2828
self.compression = compression if zstandard is not None else False
29-
self.input_type = (
30-
"pickle"
31-
if self.requests_session.get(
32-
f"{self.server_url}/meta", params={"is_pickle_allowed": ""}
33-
).json()["is_pickle_allowed"]
34-
else "msgpack"
35-
if msgpack is not None
36-
else "json"
37-
)
29+
try:
30+
self.input_type = (
31+
"pickle"
32+
if self.requests_session.get(
33+
f"{self.server_url}/meta", params={"is_pickle_allowed": ""}
34+
).json()["is_pickle_allowed"]
35+
else "msgpack"
36+
if msgpack is not None
37+
else "json"
38+
)
39+
except Exception as e:
40+
self.input_type = None
41+
3842
self.request_timeout = request_timeout
3943

4044
@property
@@ -74,6 +78,9 @@ def _decompressor(self):
7478
return self.local_storage.decompressor
7579

7680
def infer(self, data, unique_id=None, is_async=False):
81+
if self.input_type is None:
82+
raise ValueError("Could not connect to server")
83+
7784
assert isinstance(data, (list, tuple)), "Data must be of type list or tuple"
7885

7986
unique_id = str(uuid.uuid4()) if not unique_id else unique_id

Diff for: clients/python/setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
1919
AUTHOR = "BEDAPUDI PRANEETH"
2020
REQUIRES_PYTHON = ">=3.6.0"
21-
VERSION = "3.0.8"
21+
VERSION = "3.0.9"
2222

2323
# What packages are required for this module to be executed?
2424
REQUIRED = ["zstandard", "requests", "msgpack"]

0 commit comments

Comments
 (0)