Merge pull request #489 from Ukyeon/fano

Remove warning msg form nxviz
aristoteleo · May 5, 2023 · 51e2e34 · 51e2e34
2 parents 2ec448e + 5e1d4d6
commit 51e2e34
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/dynamo/plot/networks.py b/dynamo/plot/networks.py
@@ -7,8 +7,6 @@
 
 import networkx as nx
 import numpy as np
-import nxviz as nv
-import nxviz.annotate
 import pandas as pd
 from anndata import AnnData
 from matplotlib.axes import Axes
@@ -31,7 +29,7 @@ def nxvizPlot(
     save_show_or_return: Literal["save", "show", "return"] = "show",
     save_kwargs: Dict[str, Any] = {},
     **kwargs,
-) -> Union[nv.ArcPlot, nv.CircosPlot, None]:
+) -> Optional[Any]:
     """Arc or circos plot of gene regulatory network for a particular cell cluster.
 
     Args:
@@ -219,7 +217,7 @@ def arcPlot(
     save_show_or_return: Literal["save", "show", "return"] = "show",
     save_kwargs: Dict[str, Any] = {},
     **kwargs,
-) -> Optional[nv.ArcPlot]:
+) -> Optional[Any]:
     """Arc plot of gene regulatory network for a particular cell cluster.
 
     Args:
@@ -367,6 +365,11 @@ def circosPlot(
     Returns:
         the Matplotlib Axes on which the Circos plot is drawn.
     """
+    try:
+        import nxviz as nv
+    except ImportError:
+        raise ImportError("install nxviz via `pip install nxviz`.")
+
     ax = nv.circos(
         network,
         group_by=node_label_key,
@@ -402,7 +405,7 @@ def circosPlotDeprecated(
     save_show_or_return: Literal["save", "show", "return"] = "show",
     save_kwargs: Dict[str, Any] = {},
     **kwargs,
-) -> nv.CircosPlot:
+) -> Optional[Any]:
 
     """Deprecated.