Skip to content

Commit 45d7cc8

Browse files
committed
function signature consistency pass
1 parent 8c3efd3 commit 45d7cc8

1 file changed

Lines changed: 23 additions & 8 deletions

File tree

gridded/variable.py

Lines changed: 23 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -603,7 +603,7 @@ def at(
603603
If time is out of bounds of the time series, and extrapolate is False,
604604
a gridded.time.OutOfTimeRangeError is raised.
605605
"""
606-
pts, time, _hash = self._prepare_at(
606+
pts, time, _hash, kwargs = self._prepare_at(
607607
points=points,
608608
time=time,
609609
units=units,
@@ -614,8 +614,24 @@ def at(
614614
_mem=_mem,
615615
**kwargs,
616616
)
617-
value = self._compute_at(pts, time, extrapolate, _mem=_mem, _hash=_hash, **kwargs)
618-
return self._post_compute_at(value, pts, time, unmask=unmask, _mem=_mem, _hash=_hash, **kwargs)
617+
value = self._compute_at(
618+
pts,
619+
time,
620+
extrapolate,
621+
_hash=_hash,
622+
_mem=_mem,
623+
**kwargs
624+
)
625+
return self._post_compute_at(
626+
value,
627+
pts,
628+
time=time,
629+
units=units,
630+
unmask=unmask,
631+
_hash=_hash,
632+
_mem=_mem,
633+
**kwargs
634+
)
619635

620636
def _prepare_at(
621637
self,
@@ -1314,7 +1330,7 @@ def at(
13141330
gridded.time.OutOfTimeRangeError is raised.
13151331
13161332
"""
1317-
pts, time, _hash = self._prepare_at(
1333+
pts, time, _hash, kwargs = self._prepare_at(
13181334
points=points,
13191335
time=time,
13201336
units=units,
@@ -1328,9 +1344,8 @@ def at(
13281344

13291345
value = self._compute_at(
13301346
pts,
1331-
time=time,
1332-
units=units,
1333-
unmask=unmask,
1347+
time,
1348+
extrapolate,
13341349
_hash=_hash,
13351350
_mem=_mem,
13361351
**kwargs
@@ -1367,7 +1382,7 @@ def _prepare_at(
13671382
if _hash is None:
13681383
_hash = self._get_hash(pts, time)
13691384

1370-
return pts, time, _hash
1385+
return pts, time, _hash, kwargs
13711386

13721387
def _compute_at(self, points, time, extrapolate, _mem=True, _hash=None, **kwargs):
13731388
if _mem:

0 commit comments

Comments
 (0)