Remove nx

pyproject.toml

@@ -35,7 +35,6 @@ dependencies = [
     "websockets==10.3",
     "python-dateutil==2.8.0",
     "numpy==1.24.3",
-    "networkx",
     "tqdm",
     "pytest"
 ]

src/p2lab/genetic/matching.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
-import networkx as nx
+from itertools import combinations
+
+# import networkx as nx
 import numpy as np
 
 
@@ -17,23 +19,20 @@ def dense(teams: list[object]) -> np.ndarray:
             np.ndarray shape (N_edges, 2)
 
     """
-    g = nx.complete_graph(len(teams))
-
-    # Select only the node index pairs, exclude the feature dict.
-    return np.stack(g.edges(data=False))
-
-
-def mst(teams: list[object]) -> np.ndarray:
-    r"""
-    Returns a minimum spanning tree graph generated from the
-    fully connected graph over the teams.
-
-    Outputs:
-
-            np.ndarray shape (N_edges, 2)
-
-    """
-    g = nx.complete_graph(len(teams))
-    mst = nx.algorithms.tree.minimum_spanning_edges(g, data=False)
-
-    return list(mst)
+    return list(combinations(range(len(teams)), 2))
+
+
+# def mst(teams: list[object]) -> np.ndarray:
+#    r"""
+#    Returns a minimum spanning tree graph generated from the
+#    fully connected graph over the teams.
+#
+#    Outputs:
+#
+#            np.ndarray shape (N_edges, 2)
+#
+#    """
+#    g = nx.complete_graph(len(teams))
+#    mst = nx.algorithms.tree.minimum_spanning_edges(g, data=False)
+#
+#    return list(mst)