Skip to content

Commit 6efb311

Browse files
committed
use requests_cache session with retry-params when 500 errors occur
1 parent 84305e4 commit 6efb311

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
@@ -5983,10 +5983,11 @@ def test_session(self) -> None:
59835983
def test_batchdap4_downloads(protocol, batch) -> None:
59845984
"""Test that in dap4, all dimensions are downloaded at once"""
59855985
import pydap
5986-
from requests_cache import CachedSession
5986+
from pydap.net import create_session
59875987

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

@@ -6029,10 +6030,12 @@ def test_batchdap4_downloads(protocol, batch) -> None:
60296030
@requires_pydap
60306031
@network
60316032
def test_batch_warnswithdap2() -> None:
6032-
from requests_cache import CachedSession
6033+
from pydap.net import create_session
60336034

6034-
session = CachedSession()
6035+
# Create a session with pre-set retry params in pydap backend, to cache urls
6036+
session = create_session(use_cache=True, cache_kwargs={"cache_name": "debug"})
60356037
session.cache.clear()
6038+
60366039
url = "dap2://test.opendap.org/opendap/hyrax/data/nc/coads_climatology.nc"
60376040
with pytest.warns(UserWarning):
60386041
open_dataset(

xarray/tests/test_backends_datatree.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -513,13 +513,14 @@ def test_inherited_coords(self, url=simplegroup_datatree_url) -> None:
513513
| Salinity (time, Z, Y, X) float32 ...
514514
"""
515515
import pydap
516-
from requests_cache import CachedSession
516+
from pydap.net import create_session
517517

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

522+
_version_ = Version(pydap.__version__)
523+
523524
tree = open_datatree(url, engine=self.engine, session=session)
524525
assert set(tree.dims) == {"time", "Z", "nv"}
525526
assert tree["/SimpleGroup"].coords["time"].dims == ("time",)

0 commit comments

Comments
 (0)