Skip to content

Commit 579e3c4

Browse files
committed
docs fixing started
1 parent 7ae56bd commit 579e3c4

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

httomolibgpu/cupywrapper.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
try:
33
import cupy as cp
44
import nvtx
5+
import cupyx
56

67
try:
78
cp.cuda.Device(0).compute_capability
@@ -17,3 +18,4 @@
1718
import numpy as cp
1819

1920
nvtx = Mock()
21+
cupyx = Mock()

httomolibgpu/misc/morph.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,11 @@
2424
from httomolibgpu import cupywrapper
2525

2626
cp = cupywrapper.cp
27-
2827
nvtx = cupywrapper.nvtx
28+
cupyx = cupywrapper.cupyx
29+
30+
from cupyx.scipy.interpolate import interpn
31+
2932
from typing import Literal
3033

3134
__all__ = [
@@ -134,8 +137,7 @@ def data_resampler(
134137
@nvtx.annotate()
135138
def __data_resampler(
136139
data: cp.ndarray, newshape: list, axis: int = 1, interpolation: str = "linear"
137-
) -> cp.ndarray:
138-
from cupyx.scipy.interpolate import interpn
140+
) -> cp.ndarray:
139141

140142
if data.ndim != 3:
141143
raise ValueError("only 3D data is supported")

0 commit comments

Comments
 (0)