Ki-Seki commited on
Commit
94742c1
·
1 Parent(s): 6e11e8c

fix: tqdm bar

Browse files
Files changed (1) hide show
  1. 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(range(self.num_examples, self.num_data, self.save_every))
138
- for start in tqdm_bar:
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)