Request Timeout Test Fix

replicate · Nov 20, 2024 · b07e7ec · b07e7ec
1 parent 72b2933
commit b07e7ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cog/types.py b/python/cog/types.py
@@ -336,7 +336,7 @@ def __wrapped__(self) -> Any:
         except AttributeError:
             pass
         url = object.__getattribute__(self, "__url__")
-        resp = requests.get(url, stream=True, timeout=None)
+        resp = requests.get(url, stream=True, timeout=20)
         resp.raise_for_status()
         resp.raw.decode_content = True
         object.__setattr__(self, "__target__", resp.raw)