retry for input_type if None

Signed-off-by: Praneeth Bedapudi <[email protected]>

Author: bedapudi6788

clients/python/fdclient/client.py

@@ -26,21 +26,25 @@ def __init__(self, server_url, request_timeout, compression=True):
         self.local_storage = threading.local()
         self.requests_session = requests.Session()
         self.compression = compression if zstandard is not None else False
-        try:
-            self.input_type = (
-                "pickle"
-                if self.requests_session.get(
-                    f"{self.server_url}/meta", params={"is_pickle_allowed": ""}
-                ).json()["is_pickle_allowed"]
-                else "msgpack"
-                if msgpack is not None
-                else "json"
-            )
-        except Exception as e:
-            self.input_type = None
+        self._set_input_type()
 
         self.request_timeout = request_timeout
 
+    def _set_input_type(self):
+        if self.input_type is None:
+            try:
+                self.input_type = (
+                    "pickle"
+                    if self.requests_session.get(
+                        f"{self.server_url}/meta", params={"is_pickle_allowed": ""}
+                    ).json()["is_pickle_allowed"]
+                    else "msgpack"
+                    if msgpack is not None
+                    else "json"
+                )
+            except Exception as e:
+                self.input_type = None
+
     @property
     def _compressor(self):
         if self.compression is False:
@@ -79,7 +83,9 @@ def _decompressor(self):
 
     def infer(self, data, unique_id=None, is_async=False):
         if self.input_type is None:
-            raise ValueError("Could not connect to server")
+            self._set_input_type()
+            if self.input_type is None:
+                raise ValueError("Could not connect to server")
 
         assert isinstance(data, (list, tuple)), "Data must be of type list or tuple"
 

clients/python/setup.py

@@ -18,7 +18,7 @@
 EMAIL = "[email protected]"
 AUTHOR = "BEDAPUDI PRANEETH"
 REQUIRES_PYTHON = ">=3.6.0"
-VERSION = "3.0.9"
+VERSION = "3.1.0"
 
 # What packages are required for this module to be executed?
 REQUIRED = ["zstandard", "requests", "msgpack"]