Skip to content

Commit c851a79

Browse files
authored
add documentation for get (#371)
1 parent da8f291 commit c851a79

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/rapids_singlecell/get/_anndata.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ def anndata_to_GPU(
7575
return adata
7676

7777

78-
def X_to_GPU(X: CPU_ARRAY_TYPE, warning: str = "X") -> GPU_ARRAY_TYPE:
78+
def X_to_GPU(
79+
X: CPU_ARRAY_TYPE | DaskArray, warning: str = "X"
80+
) -> GPU_ARRAY_TYPE | DaskArray:
7981
"""
8082
Transfers matrices and arrays to the GPU
8183
@@ -151,7 +153,7 @@ def anndata_to_CPU(
151153
return adata
152154

153155

154-
def X_to_CPU(X: GPU_ARRAY_TYPE) -> CPU_ARRAY_TYPE:
156+
def X_to_CPU(X: GPU_ARRAY_TYPE | DaskArray) -> CPU_ARRAY_TYPE | DaskArray:
155157
"""
156158
Transfers matrices and arrays from the GPU
157159

0 commit comments

Comments
 (0)