chore

Author: calad0i

hls4ml/backends/fpga/passes/clone.py

@@ -86,13 +86,13 @@ def transform(self, model, node):
             out_var = node.get_output_variable(output)
             attrs = {'size': np.prod(out_var.shape)}
 
-            i0 = 1
+            init_stream_idx = 1
             if in_output:
                 # If the value is used as output, add one extra stream
                 idx = node.model.outputs.index(node.name)
                 node.model.outputs[idx] = node.name + '_cpy1'
-                i0 = 2
-            for i, layer in enumerate(output_map[output], i0):
+                init_stream_idx = 2
+            for i, layer in enumerate(output_map[output], init_stream_idx):
                 idx = layer.inputs.index(output)
                 layer.inputs[idx] = output + f'_cpy{i}'
 

hls4ml/model/graph.py

@@ -520,15 +520,18 @@ def insert_node(self, node, before=None, input_idx=0):
     def remove_node(self, node, rewire=True):
         """Remove a node from a graph.
 
-        By default, this function can connect the outputs of previous node to the input of next one. If the removed node has multiple inputs/outputs tensors, an exception is raised.
+        By default, this function can connect the outputs of previous
+        node to the input of next one. If the removed node has multiple
+        inputs/outputs tensors, an exception is raised.
 
         Args:
-            node (Layer): The node to remove
-            rewire (bool, optional): Deprecated, no effect
+            node (Layer): The node to remove rewire (bool, optional):
+            Deprecated, no effect
 
         Raises:
-            Exception: If an attempt is made to rewire a node with multiple inputs/outputs.
-        """  # noqa: E501
+            Exception: If an attempt is made to rewire a node with
+            multiple inputs/outputs.
+        """
 
         inputs = [inp for inp in node.inputs if inp]
         outputs = [outp for outp in node.outputs if outp]