We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent da8f291 commit c851a79Copy full SHA for c851a79
1 file changed
src/rapids_singlecell/get/_anndata.py
@@ -75,7 +75,9 @@ def anndata_to_GPU(
75
return adata
76
77
78
-def X_to_GPU(X: CPU_ARRAY_TYPE, warning: str = "X") -> GPU_ARRAY_TYPE:
+def X_to_GPU(
79
+ X: CPU_ARRAY_TYPE | DaskArray, warning: str = "X"
80
+) -> GPU_ARRAY_TYPE | DaskArray:
81
"""
82
Transfers matrices and arrays to the GPU
83
@@ -151,7 +153,7 @@ def anndata_to_CPU(
151
153
152
154
155
-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:
157
158
Transfers matrices and arrays from the GPU
159
0 commit comments