fix: graph subclassing

need to be careful here...
arangodb · Apr 27, 2024 · 681eb26 · 681eb26
1 parent 261a423
commit 681eb26
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/nx_arangodb/classes/digraph.py b/nx_arangodb/classes/digraph.py
@@ -3,13 +3,14 @@
 import networkx as nx
 
 import nx_arangodb as nxadb
+from nx_arangodb.classes.graph import Graph
 
 networkx_api = nxadb.utils.decorators.networkx_class(nx.DiGraph)
 
 __all__ = ["DiGraph"]
 
 
-class DiGraph(nx.DiGraph):
+class DiGraph(nx.DiGraph, Graph):
     __networkx_backend__: ClassVar[str] = "arangodb"  # nx >=3.2
     __networkx_plugin__: ClassVar[str] = "arangodb"  # nx <3.2
 

diff --git a/nx_arangodb/classes/multidigraph.py b/nx_arangodb/classes/multidigraph.py
@@ -3,13 +3,15 @@
 import networkx as nx
 
 import nx_arangodb as nxadb
+from nx_arangodb.classes.digraph import DiGraph
+from nx_arangodb.classes.multigraph import MultiGraph
 
 networkx_api = nxadb.utils.decorators.networkx_class(nx.MultiDiGraph)
 
 __all__ = ["MultiDiGraph"]
 
 
-class MultiDiGraph(nx.MultiDiGraph):
+class MultiDiGraph(nx.MultiDiGraph, MultiGraph, DiGraph):
     __networkx_backend__: ClassVar[str] = "arangodb"  # nx >=3.2
     __networkx_plugin__: ClassVar[str] = "arangodb"  # nx <3.2
 

diff --git a/nx_arangodb/classes/multigraph.py b/nx_arangodb/classes/multigraph.py
@@ -3,13 +3,14 @@
 import networkx as nx
 
 import nx_arangodb as nxadb
+from nx_arangodb.classes.graph import Graph
 
 networkx_api = nxadb.utils.decorators.networkx_class(nx.MultiGraph)
 
 __all__ = ["MultiGraph"]
 
 
-class MultiGraph(nx.MultiGraph):
+class MultiGraph(nx.MultiGraph, Graph):
     __networkx_backend__: ClassVar[str] = "arangodb"  # nx >=3.2
     __networkx_plugin__: ClassVar[str] = "arangodb"  # nx <3.2