Implement AnyRandom

CSOgroup · Jul 2, 2024 · 5eb35ea · 5eb35ea
1 parent 44106ad
commit 5eb35ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/cellcharter/_utils.py b/src/cellcharter/_utils.py
@@ -2,6 +2,10 @@
 
 from typing import Union
 
+import numpy as np
+
+AnyRandom = Union[int, np.random.RandomState, None]
+
 
 def str2list(value: Union[str, list]) -> list:
     """Check whether value is a string. If so, converts into a list containing value."""

diff --git a/src/cellcharter/tl/_gmm.py b/src/cellcharter/tl/_gmm.py
@@ -14,9 +14,10 @@
 from pycave.bayes.gmm.lightning_module import GaussianMixtureLightningModule
 from pycave.bayes.gmm.model import GaussianMixtureModel
 from pytorch_lightning import Trainer
-from scanpy._utils import AnyRandom
 from sklearn.cluster import KMeans, MiniBatchKMeans
 
+from .._utils import AnyRandom
+
 logger = logging.getLogger(__name__)