Skip to content

Commit eb3fca5

Browse files
committed
use requests_cache session with retry-params when 500 errors occur
1 parent 6c354ca commit eb3fca5

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
@@ -6472,10 +6472,11 @@ def test_session(self) -> None:
64726472
def test_batchdap4_downloads(protocol, batch) -> None:
64736473
"""Test that in dap4, all dimensions are downloaded at once"""
64746474
import pydap
6475-
from requests_cache import CachedSession
6475+
from pydap.net import create_session
64766476

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

@@ -6518,10 +6519,12 @@ def test_batchdap4_downloads(protocol, batch) -> None:
65186519
@requires_pydap
65196520
@network
65206521
def test_batch_warnswithdap2() -> None:
6521-
from requests_cache import CachedSession
6522+
from pydap.net import create_session
65226523

6523-
session = CachedSession()
6524+
# Create a session with pre-set retry params in pydap backend, to cache urls
6525+
session = create_session(use_cache=True, cache_kwargs={"cache_name": "debug"})
65246526
session.cache.clear()
6527+
65256528
url = "dap2://test.opendap.org/opendap/hyrax/data/nc/coads_climatology.nc"
65266529
with pytest.warns(UserWarning):
65276530
open_dataset(

xarray/tests/test_backends_datatree.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -641,13 +641,14 @@ def test_inherited_coords(self, url=simplegroup_datatree_url) -> None:
641641
| Salinity (time, Z, Y, X) float32 ...
642642
"""
643643
import pydap
644-
from requests_cache import CachedSession
644+
from pydap.net import create_session
645645

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

650+
_version_ = Version(pydap.__version__)
651+
651652
tree = open_datatree(url, engine=self.engine, session=session)
652653
assert set(tree.dims) == {"time", "Z", "nv"}
653654
assert tree["/SimpleGroup"].coords["time"].dims == ("time",)

0 commit comments

Comments
 (0)