Fix Client doesn't recover from remote connection resets (#2129)

Author: avishniakov

examples/nlp-case/run.py

@@ -67,7 +67,7 @@
 @click.option(
     "--no-cache",
     is_flag=True,
-    default=False,
+    default=True,
     help="Disable caching for the pipeline run.",
 )
 @click.option(

src/zenml/zen_stores/rest_zen_store.py

@@ -32,6 +32,7 @@
 import requests
 import urllib3
 from pydantic import BaseModel, root_validator, validator
+from requests.adapters import HTTPAdapter, Retry
 
 import zenml
 from zenml.analytics import source_context
@@ -3183,6 +3184,9 @@ def session(self) -> requests.Session:
                 )
 
             self._session = requests.Session()
+            retries = Retry(backoff_factor=0.1, connect=5)
+            self._session.mount("https://", HTTPAdapter(max_retries=retries))
+            self._session.mount("http://", HTTPAdapter(max_retries=retries))
             self._session.verify = self.config.verify_ssl
             token = self._get_auth_token()
             self._session.headers.update({"Authorization": "Bearer " + token})