From cb02e91831d6d16711459100fe4cccdb484d0ae0 Mon Sep 17 00:00:00 2001 From: mattephi Date: Tue, 17 Sep 2024 19:08:10 +0900 Subject: [PATCH] fix: no tqdm --- jaxadi/_stages.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/jaxadi/_stages.py b/jaxadi/_stages.py index bd6f39e..73bfc2a 100644 --- a/jaxadi/_stages.py +++ b/jaxadi/_stages.py @@ -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 @@ -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]