Skip to content

Commit

Permalink
[Lint] run pre-commit for PR#113
Browse files Browse the repository at this point in the history
  • Loading branch information
maltanar committed Dec 16, 2024
1 parent e106d09 commit 5d246e8
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/qonnx/transformation/create_generic_partitions.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,7 +202,7 @@ def partitioning_func(node):
if len(candidates) == 0:
return -1
assert len(candidates) == 1, f"single node assigned to multiple partitions: {candidates}"
return candidates[0][0] # partition_id
return candidates[0][0] # partition_id

# apply partitioning
model = model.transform(PartitionFromLambda(partitioning_func, self.partition_dir))
Expand Down

0 comments on commit 5d246e8

Please sign in to comment.