fix: tqdm bar
Browse files- autotab.py +3 -3
autotab.py
CHANGED
@@ -134,9 +134,9 @@ class AutoTab:
|
|
134 |
self.num_data = len(self.data)
|
135 |
self.num_examples = len(self.data.dropna(subset=self.output_fields))
|
136 |
|
137 |
-
tqdm_bar = tqdm(
|
138 |
-
for start in
|
139 |
-
tqdm_bar.update(start)
|
140 |
end = min(start + self.save_every, self.num_data)
|
141 |
try:
|
142 |
self.batch_prediction(start, end)
|
|
|
134 |
self.num_data = len(self.data)
|
135 |
self.num_examples = len(self.data.dropna(subset=self.output_fields))
|
136 |
|
137 |
+
tqdm_bar = tqdm(total=self.num_data - self.num_examples, leave=False)
|
138 |
+
for start in range(self.num_examples, self.num_data, self.save_every):
|
139 |
+
tqdm_bar.update(min(self.save_every, self.num_data - start))
|
140 |
end = min(start + self.save_every, self.num_data)
|
141 |
try:
|
142 |
self.batch_prediction(start, end)
|