From 9585f550b32d7591f63273aab9483fda9cc37920 Mon Sep 17 00:00:00 2001
From: Will Jones <willjones127@gmail.com>
Date: Fri, 15 Sep 2023 23:21:34 -0400
Subject: [PATCH] fix: increase S3 timeouts (#494)

Closes #493
---
 python/lancedb/util.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/lancedb/util.py b/python/lancedb/util.py
index 1abd8dc7..15b1d427 100644
--- a/python/lancedb/util.py
+++ b/python/lancedb/util.py
@@ -70,7 +70,11 @@ def fs_from_uri(uri: str) -> Tuple[pa_fs.FileSystem, str]:
     Get a PyArrow FileSystem from a URI, handling extra environment variables.
     """
     if get_uri_scheme(uri) == "s3":
-        fs = pa_fs.S3FileSystem(endpoint_override=os.environ.get("AWS_ENDPOINT"))
+        fs = pa_fs.S3FileSystem(
+            endpoint_override=os.environ.get("AWS_ENDPOINT"),
+            request_timeout=30,
+            connect_timeout=30,
+        )
         path = get_uri_location(uri)
         return fs, path