fix: retry env to be None type

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Oct 31, 2022 · fe205c5 · fe205c5
1 parent 92461d7
commit fe205c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/bentoml/_internal/runner/runner_handle/remote.py b/src/bentoml/_internal/runner/runner_handle/remote.py
@@ -171,7 +171,8 @@ async def async_run_method(
             ) as resp:
                 body = await resp.read()
         except aiohttp.ClientOSError as e:
-            if os.getenv("BENTOML_RETRY_RUNNER_REQUESTS").lower() == "true":
+            retry_env = os.environ.get("BENTOML_RETRY_RUNNER_REQUESTS")
+            if retry_env is not None and retry_env.lower() == "true":
                 try:
                     # most likely the TCP connection has been closed; retry after reconnecting
                     await self._reset_client()