just tqdm notebook for fit

arjunsavel · Feb 15, 2024 · a4598ec · a4598ec
1 parent 57be7fc
commit a4598ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cortecs/fit/fit.py b/src/cortecs/fit/fit.py
@@ -6,7 +6,7 @@
 from multiprocessing import Pool
 
 import numpy as np
-from tqdm.autonotebook import tqdm
+from tqdm.notebook import tqdm
 
 from cortecs.fit.fit_neural_net import *
 from cortecs.fit.fit_pca import *