Skip to content

Commit 2d5f10c

Browse files
committed
use requests_cache session with retry-params when 500 errors occur
1 parent 525e7c2 commit 2d5f10c

File tree

3 files changed

+12
-9
lines changed

3 files changed

+12
-9
lines changed

xarray/backends/pydap_.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,6 @@ def __init__(
117117
"""
118118
self.dataset = dataset
119119
self.group = group
120-
self.session = session
121120
self._batch = batch
122121
self._batch_done = False
123122
self._array_cache = {} # holds 1D dimension data

xarray/tests/test_backends.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6456,10 +6456,11 @@ def test_session(self) -> None:
64566456
def test_batchdap4_downloads(protocol, batch) -> None:
64576457
"""Test that in dap4, all dimensions are downloaded at once"""
64586458
import pydap
6459-
from requests_cache import CachedSession
6459+
from pydap.net import create_session
64606460

64616461
_version_ = Version(pydap.__version__)
6462-
session = CachedSession(cache_name="debug") # so that urls are cached
6462+
# Create a session with pre-set params in pydap backend, to cache urls
6463+
session = create_session(use_cache=True, cache_kwargs={"cache_name": "debug"})
64636464
session.cache.clear()
64646465
url = "https://test.opendap.org/opendap/hyrax/data/nc/coads_climatology.nc"
64656466

@@ -6502,10 +6503,12 @@ def test_batchdap4_downloads(protocol, batch) -> None:
65026503
@requires_pydap
65036504
@network
65046505
def test_batch_warnswithdap2() -> None:
6505-
from requests_cache import CachedSession
6506+
from pydap.net import create_session
65066507

6507-
session = CachedSession()
6508+
# Create a session with pre-set retry params in pydap backend, to cache urls
6509+
session = create_session(use_cache=True, cache_kwargs={"cache_name": "debug"})
65086510
session.cache.clear()
6511+
65096512
url = "dap2://test.opendap.org/opendap/hyrax/data/nc/coads_climatology.nc"
65106513
with pytest.warns(UserWarning):
65116514
open_dataset(

xarray/tests/test_backends_datatree.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -635,13 +635,14 @@ def test_inherited_coords(self, url=simplegroup_datatree_url) -> None:
635635
| Salinity (time, Z, Y, X) float32 ...
636636
"""
637637
import pydap
638-
from requests_cache import CachedSession
638+
from pydap.net import create_session
639639

640-
_version_ = Version(pydap.__version__)
641-
642-
session = CachedSession(cache_name="debug") # so that urls are cached
640+
# Create a session with pre-set retry params in pydap backend, to cache urls
641+
session = create_session(use_cache=True, cache_kwargs={"cache_name": "debug"})
643642
session.cache.clear()
644643

644+
_version_ = Version(pydap.__version__)
645+
645646
tree = open_datatree(url, engine=self.engine, session=session)
646647
assert set(tree.dims) == {"time", "Z", "nv"}
647648
assert tree["/SimpleGroup"].coords["time"].dims == ("time",)

0 commit comments

Comments
 (0)