Skip to content

Commit ab30e48

Browse files
committed
added unit mismatch check to curvature
1 parent 33d0f5f commit ab30e48

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

xrspatial/curvature.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@ class cupy(object):
2121
from numba import cuda
2222

2323
# local modules
24-
from xrspatial.utils import (ArrayTypeFunctionMapping, cuda_args, get_dataarray_resolution, ngjit)
24+
from xrspatial.utils import ArrayTypeFunctionMapping
25+
from xrspatial.utils import cuda_args
26+
from xrspatial.utils import get_dataarray_resolution
27+
from xrspatial.utils import ngjit
28+
from xrspatial.utils import warn_if_unit_mismatch
2529

2630

2731
@ngjit
@@ -220,6 +224,7 @@ def curvature(agg: xr.DataArray,
220224
Attributes:
221225
res: (10, 10)
222226
"""
227+
warn_if_unit_mismatch(agg)
223228

224229
cellsize_x, cellsize_y = get_dataarray_resolution(agg)
225230
cellsize = (cellsize_x + cellsize_y) / 2

0 commit comments

Comments
 (0)