ENH: Use the new RetrySession context manager. [skip ci]

hyriver · Sep 22, 2023 · ae4013a · ae4013a
1 parent 2031c31
commit ae4013a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pygeohydro/pygeohydro.py b/pygeohydro/pygeohydro.py
@@ -144,7 +144,7 @@ def ssebopeta_bycoords(
     f_list = helpers.get_ssebopeta_urls(dates)
     session = RetrySession()
 
-    with patch("socket.has_ipv6", False):
+    with patch("socket.has_ipv6", False), RetrySession() as session:
 
         def _ssebop(url: str) -> list[npt.NDArray[np.float64]]:
             r = session.get(url)
@@ -216,8 +216,7 @@ def ssebopeta_bygeom(
         raise InputTypeError("geometry", "(Multi)Polygon or tuple of length 4") from ex
 
     gtiff2xarray = tlz.partial(geoutils.gtiff2xarray, geometry=geometry, geo_crs=geo_crs)
-    session = RetrySession()
-    with patch("socket.has_ipv6", False):
+    with patch("socket.has_ipv6", False), RetrySession() as session:
 
         def _ssebop(t: pd.Timestamp, url: str) -> xr.DataArray:
             resp = session.get(url)