Hotfix progress bar.

wilhelm-lab · May 13, 2024 · 06421df · 06421df
1 parent 3383dc2
commit 06421df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clients/python/koinapy/grpc.py b/clients/python/koinapy/grpc.py
@@ -517,7 +517,7 @@ def __predict_semi_async(self, data, debug=False, disable_progress_bar=False):
         results = []
         data_subsets = list(self.__slice_dict(data, self.batchsize * 10))
         pbar = tqdm(
-            total=len(data_subsets) * 10,
+            total=ceil(next(iter(data.values())).shape[0] / self.batchsize),
             desc=f"{self.model_name}:",
             disable=disable_progress_bar,
         )