organize types

OmicsML · Feb 4, 2024 · f6760db · f6760db
1 parent 821ad7b
commit f6760db
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/dance/transforms/normalize.py b/dance/transforms/normalize.py
@@ -1,7 +1,5 @@
 import os
 from multiprocessing import Manager, Pool
-from numbers import Number
-from typing import Iterable
 
 import anndata as ad
 import numpy as np
@@ -17,7 +15,7 @@
 from dance.registry import register_preprocessor
 from dance.transforms.base import BaseTransform
 from dance.transforms.interface import AnnDataTransform
-from dance.typing import Dict, List, Literal, NormMode, Optional, Union
+from dance.typing import Dict, Iterable, List, Literal, NormMode, Number, Optional, Union
 from dance.utils.matrix import normalize
 
 

diff --git a/dance/typing.py b/dance/typing.py
@@ -2,11 +2,13 @@
 
 import os
 from logging import Logger
+from numbers import Number
 from typing import (
     TYPE_CHECKING,
     Any,
     Callable,
     Dict,
+    Iterable,
     Iterator,
     List,
     Literal,
@@ -48,12 +50,14 @@
     "FeatType",
     "FileExistHandle",
     "GeneSummaryMode",
+    "Iterable",
     "Iterator",
     "List",
     "LogLevel",
     "Logger",
     "Mapping",
     "NormMode",
+    "Number",
     "Optional",
     "ReturnedFeat",
     "Sequence",