test TypeAlias

single-cell-data · Aug 25, 2024 · 5dc6cfd · 5dc6cfd
1 parent 0e32457
commit 5dc6cfd
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/other_packages/python/tiledbsoma_ml/src/tiledbsoma_ml/pytorch.py b/other_packages/python/tiledbsoma_ml/src/tiledbsoma_ml/pytorch.py
@@ -15,7 +15,6 @@
 from itertools import islice
 from math import ceil
 from typing import (
-    TYPE_CHECKING,
     Any,
     Dict,
     Iterable,
@@ -46,10 +45,11 @@
 _T = TypeVar("_T")
 _T_co = TypeVar("_T_co", covariant=True)
 
-if TYPE_CHECKING:
-    NDArrayNumber: TypeAlias = npt.NDArray[np.number[Any]]
-else:
-    NDArrayNumber: TypeAlias = np.ndarray
+# if TYPE_CHECKING:
+#     NDArrayNumber: TypeAlias = npt.NDArray[np.number[Any]]
+# else:
+#     NDArrayNumber: TypeAlias = np.ndarray
+NDArrayNumber: TypeAlias = npt.NDArray[np.number[Any]]
 XObsDatum: TypeAlias = Tuple[NDArrayNumber, pd.DataFrame]
 XObsNpDatum: TypeAlias = Tuple[NDArrayNumber, NDArrayNumber]
 XObsTensorDatum: TypeAlias = Tuple[torch.Tensor, torch.Tensor]