fix import mellon error

Starlitnightly · Nov 16, 2023 · b0ba2a8 · b0ba2a8
1 parent 78fd560
commit b0ba2a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/omicverse/palantir/utils.py b/omicverse/palantir/utils.py
@@ -8,7 +8,7 @@
 
 from scipy.sparse import csr_matrix, find, issparse, hstack
 from scipy.sparse.linalg import eigs
-import mellon
+#import mellon
 import scanpy as sc
 
 from .core import run_palantir
@@ -222,7 +222,7 @@ def run_density(
     # Set the default arguments for mellon.DensityEstimator
     mellon_args = dict()
     mellon_args.update(kwargs)
-
+    import mellon
     dest = mellon.DensityEstimator(**mellon_args)
     log_density = np.asarray(dest.fit_predict(X))
 
@@ -285,6 +285,7 @@ def run_density_evaluation(
         )
 
     X = out_ad.obsm[repr_key]
+    import mellon
     predictor = mellon.Predictor.from_dict(in_ad.uns[predictor_key])
     log_density = predictor(X)
     out_ad.obs[density_key] = log_density

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "omicverse"
-version = "1.5.2"
+version = "1.5.3"
 description = "OmicVerse: A single pipeline for exploring the entire transcriptome universe"
 readme = "README.md"
 requires-python = ">=3.8"