From 89c89a64dda73521301b446ed4052440e27cdeb7 Mon Sep 17 00:00:00 2001
From: cwognum <caswognum@outlook.com>
Date: Thu, 28 Mar 2024 16:27:31 -0400
Subject: [PATCH] Fix a bug in Dataset.cache() and the timeout for the Hub
 client

---
 polaris/dataset/_dataset.py | 1 +
 polaris/hub/client.py       | 1 +
 2 files changed, 2 insertions(+)

diff --git a/polaris/dataset/_dataset.py b/polaris/dataset/_dataset.py
index 5200b2de..bb1db11b 100644
--- a/polaris/dataset/_dataset.py
+++ b/polaris/dataset/_dataset.py
@@ -366,6 +366,7 @@ def cache(self, cache_dir: Optional[str] = None) -> str:
 
         if self.zarr_root_path is not None:
             self.zarr_root_path = fs.join(self.cache_dir, "data.zarr")
+            self._zarr_root = None
 
         if not self._has_been_cached:
             self._has_been_cached = True
diff --git a/polaris/hub/client.py b/polaris/hub/client.py
index 9f7e2292..b746b39d 100644
--- a/polaris/hub/client.py
+++ b/polaris/hub/client.py
@@ -115,6 +115,7 @@ def __init__(
             # httpx.Client
             base_url=settings.api_url,
             verify=verify,
+            timeout=self.settings.default_timeout,
             # Extra
             **kwargs,
         )