diff --git a/floogen/model/graph.py b/floogen/model/graph.py index e9424030..3544e613 100644 --- a/floogen/model/graph.py +++ b/floogen/model/graph.py @@ -194,6 +194,7 @@ def add_nodes_as_tree( tree: List[int], node_type: str, edge_type: str, + *, lvl: int = 0, node_obj=None, edge_obj=None, @@ -213,7 +214,14 @@ def add_nodes_as_tree( node, parent, type=edge_type, obj=edge_obj, src_dir=None, dst_dir=None ) self.add_nodes_as_tree( - node, tree, node_type, edge_type, lvl + 1, node_obj, edge_obj, connect + node, + tree, + node_type, + edge_type, + lvl=lvl + 1, + node_obj=node_obj, + edge_obj=edge_obj, + connect=connect, ) def add_nodes_as_array( @@ -221,6 +229,7 @@ def add_nodes_as_array( name: str, array: Tuple[int], node_type: str, + *, edge_type: str = "", node_obj=None, edge_obj=None, diff --git a/util/gen_jobs.py b/util/gen_jobs.py index 877638fb..4a2a669c 100755 --- a/util/gen_jobs.py +++ b/util/gen_jobs.py @@ -38,6 +38,7 @@ def gen_job_str( length: int, src_addr: int, dst_addr: int, + *, max_src_burst_size: int = 256, max_dst_burst_size: int = 256, r_aw_decouple: bool = False, @@ -101,7 +102,7 @@ def gen_nw_chimney2chimney_traffic( bidir: bool, out_dir: str ): - # pylint: disable=too-many-arguments + # pylint: disable=too-many-arguments, too-many-positional-arguments """Generate Narrow Wide Chimney to Chimney traffic.""" num_masters = 2 for i in range(num_masters): @@ -131,7 +132,7 @@ def gen_mesh_traffic( out_dir: str, **_kwargs ): - # pylint: disable=too-many-arguments, too-many-locals, too-many-branches, too-many-statements + # pylint: disable=too-many-arguments, too-many-locals, too-many-branches, too-many-statements, too-many-positional-arguments """Generate Mesh traffic.""" for x in range(0, NUM_X): for y in range(0, NUM_Y):