Added numpy and linted with ruff

cvanelteren · Sep 26, 2023 · d169a5d · d169a5d
1 parent 813a55f
commit d169a5d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/networkx/drawing/layout.py b/networkx/drawing/layout.py
@@ -1249,8 +1249,10 @@ def arc_layout(G: nx.Graph, subset_key="layer", radius=1, rotation=0) -> dict:
     >>> G = nx.multipartite_graph(*subset_sizes)
     >>> pos = nx.arf_layout(G, subset_key = "layer")
     """
+    import numpy as np
+
     attrs = nx.get_node_attributes(G, subset_key)
-    categories = {v for v in attrs.values()}
+    categories = set(attrs.values())
     angles = np.linspace(0, 2 * np.pi, len(categories), 0) + rotation
     pos = {}
     for category, angle in zip(categories, angles):