Skip to content

Commit

Permalink
style: formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
sokovninn committed Oct 11, 2024
1 parent efaf3dd commit b3a0468
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion luxonis_train/core/utils/archive_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -188,6 +188,7 @@ def _get_head_outputs(

return output_names


def get_heads(
cfg: Config,
outputs: list[dict],
Expand Down Expand Up @@ -226,7 +227,7 @@ def get_heads(
head_outputs = _get_head_outputs(
outputs, node_alias, node_name
)

if node_alias in head_names:
curr_head_name = f"{node_alias}_{len(head_names)}" # add suffix if name is already present
else:
Expand Down

0 comments on commit b3a0468

Please sign in to comment.