Use enumerate

b_asic/research/interleaver.py

@@ -46,18 +46,18 @@ def generate_random_interleaver(
     ProcessCollection
 
     """
-    inputorder = list(range(size))
-    outputorder = inputorder[:]
-    random.shuffle(outputorder)
-    inputorder, outputorder = _insert_delays(
-        inputorder, outputorder, min_lifetime, cyclic
+    inputorders = list(range(size))
+    outputorders = inputorders[:]
+    random.shuffle(outputorders)
+    inputorders, outputorders = _insert_delays(
+        inputorders, outputorders, min_lifetime, cyclic
     )
     return ProcessCollection(
         {
-            PlainMemoryVariable(inputorder[i], 0, {0: outputorder[i] - inputorder[i]})
-            for i in range(len(inputorder))
+            PlainMemoryVariable(inputorder, 0, {0: outputorders[i] - inputorder})
+            for i, inputorder in enumerate(inputorders)
         },
-        len(inputorder),
+        len(inputorders),
         cyclic,
     )
 

b_asic/signal_flow_graph.py

@@ -738,9 +738,7 @@ class SFG(AbstractOperation):
 
         # Find all first iter output ports for precedence
         first_iter_ports = [
-            op.output(i)
-            for op in (no_input_ops + delay_ops)
-            for i in range(op.output_count)
+            output for op in (no_input_ops + delay_ops) for output in op.outputs
         ]
 
         self._precedence_list = self._traverse_for_precedence_list(first_iter_ports)
@@ -756,8 +754,7 @@ class SFG(AbstractOperation):
         pg.attr(rankdir="LR")
 
         # Creates nodes for each output port in the precedence list
-        for i in range(len(p_list)):
-            ports = p_list[i]
+        for i, ports in enumerate(p_list):
             with pg.subgraph(name=f"cluster_{i}") as sub:
                 sub.attr(label=f"N{i}")
                 for port in ports:
@@ -779,8 +776,7 @@ class SFG(AbstractOperation):
                         )
         # Creates edges for each output port and creates nodes for each operation
         # and edges for them as well
-        for i in range(len(p_list)):
-            ports = p_list[i]
+        for i, ports in enumerate(p_list):
             for port in ports:
                 source_label = port.operation.graph_id
                 node_node = port.name