Skip to content

Commit

Permalink
fix: no tqdm
Browse files Browse the repository at this point in the history
  • Loading branch information
mattephi committed Sep 17, 2024
1 parent 265b84c commit 7296a47
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion jaxadi/_stages.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ def squeeze(stages: List[Stage], num_threads=1) -> List[Stage]:
for i, stage in enumerate(stages):
if len(stage.ops) != 0 and stage.ops[0].op == OP_OUTPUT:
working_stages.append((i, stage))
for i in tqdm(range(len(working_stages))):
for i in range(len(working_stages)):
i, stage = working_stages[i]
stage.value = recursive_subs(stages, i)
new_stages.append(stage)
Expand Down

0 comments on commit 7296a47

Please sign in to comment.