Merge pull request #34 from MHenderson/30-remove-some-unnecessary-code

Remove a pointless list expression
MHenderson · Dec 9, 2024 · ca73b8d · ca73b8d
2 parents 59675e5 + 275800a
commit ca73b8d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vizing"
-version = "0.2.1"
+version = "0.2.1.9000"
 description = "Constraint-based list-colouring in Python."
 authors = ["Matthew Henderson <[email protected]>"]
 readme = "README.md"

diff --git a/src/vizing/__init__.py b/src/vizing/__init__.py
@@ -40,8 +40,8 @@ def edge_list_colouring_problem(G, directed = False):
     P.addVariable(edge, G.edges[edge]['permissible'])
   if(directed):
     for node in G.nodes():
-      P.addConstraint(ct.AllDifferentConstraint(), [x for x in G.out_edges(node)])
-      P.addConstraint(ct.AllDifferentConstraint(), [x for x in G.in_edges(node)])
+      P.addConstraint(ct.AllDifferentConstraint(), G.out_edges(node))
+      P.addConstraint(ct.AllDifferentConstraint(), G.in_edges(node))
     else:
       for node in G.nodes():
         P.addConstraint(ct.AllDifferentConstraint(), [tuple(sorted(x)) for x in G.edges(node)])