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 75bbc9e commit cb02e91
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions jaxadi/_stages.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
from ._ops import OP_JAX_VALUE_DICT
from casadi import OP_CONST, OP_INPUT, OP_OUTPUT, OP_SQ, Function
import re
from tqdm import tqdm
from multiprocessing import Pool, cpu_count


Expand Down Expand Up @@ -72,7 +71,7 @@ def stage_generator(func: Function) -> str:
const_instr = [func.instruction_constant(i) for i in range(n_instr)]

stages = []
for k in tqdm(range(n_instr)):
for k in range(n_instr):
op = operations[k]
o_idx = output_idx[k]
i_idx = input_idx[k]
Expand Down

0 comments on commit cb02e91

Please sign in to comment.