fix: rm nx dep

jaxleyverse · Dec 3, 2024 · 7f4525a · 7f4525a
1 parent 40b2ba6
commit 7f4525a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/jaxley/modules/base.py b/jaxley/modules/base.py
@@ -2118,9 +2118,13 @@ def vis(
         """
         res = 100 if "resolution" not in kwargs else kwargs.pop("resolution")
         if "comp" in type.lower():
-            return plot_comps(self, dims=dims, ax=ax, color=color, resolution=res, **kwargs)
+            return plot_comps(
+                self, dims=dims, ax=ax, color=color, resolution=res, **kwargs
+            )
         if "morph" in type.lower():
-            return plot_morph(self, dims=dims, ax=ax, color=color, resolution=res, **kwargs)
+            return plot_morph(
+                self, dims=dims, ax=ax, color=color, resolution=res, **kwargs
+            )
 
         assert not np.any(
             [np.isnan(xyzr[:, dims]).all() for xyzr in self.xyzr]

diff --git a/jaxley/modules/network.py b/jaxley/modules/network.py
@@ -7,7 +7,6 @@
 from warnings import warn
 
 import jax.numpy as jnp
-import networkx as nx
 import numpy as np
 import pandas as pd
 from jax import vmap