Skip to content

Commit

Permalink
lint: Python sources
Browse files Browse the repository at this point in the history
  • Loading branch information
fischeti committed Sep 28, 2024
1 parent 8d09b86 commit 202d212
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 3 deletions.
11 changes: 10 additions & 1 deletion floogen/model/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand All @@ -213,14 +214,22 @@ 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(
self,
name: str,
array: Tuple[int],
node_type: str,
*,
edge_type: str = "",
node_obj=None,
edge_obj=None,
Expand Down
5 changes: 3 additions & 2 deletions util/gen_jobs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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):
Expand Down Expand Up @@ -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):
Expand Down

0 comments on commit 202d212

Please sign in to comment.