NetCon can set multiple weights if an iterator is provided.

dbbs-lab · Oct 11, 2020 · 803cd4f · 803cd4f
1 parent 0b16107
commit 803cd4f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/patch/interpreter.py b/patch/interpreter.py
@@ -70,7 +70,11 @@ def NetCon(self, source, target, *args, **kwargs):
         # Set the weight, delay and threshold independently
         for k, v in setters.items():
             if k == "weight":
-                connection.weight[0] = v
+                if hasattr(type(v), "__iter__"):  # pragma: nocover
+                    for i, w in enumerate(v):
+                        connection.weight[i] = w
+                else:
+                    connection.weight[0] = v
             else:
                 setattr(connection, k, v)
         # Have the NetCon reference source and target