version dependent networkx (#266)

CAVEconnectome · Nov 14, 2024 · 069fac7 · 069fac7
1 parent acbc7c8
commit 069fac7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/caveclient/chunkedgraph.py b/caveclient/chunkedgraph.py
@@ -10,6 +10,7 @@
 import numpy as np
 import pandas as pd
 import pytz
+from packaging.version import Version
 
 from .auth import AuthClient
 from .base import (
@@ -996,7 +997,10 @@ def get_lineage_graph(
             ]
 
         if as_nx_graph:
-            return nx.node_link_graph(r, edges="links")
+            if Version(nx.__version__) >= Version("3.4"):
+                return nx.node_link_graph(r, edges="links")
+            else:
+                return nx.node_link_graph(r)
         else:
             return r