[Prediction] Fixed display loader

YanSte · Aug 9, 2023 · e079f5c · e079f5c
1 parent 9089bf1
commit e079f5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/skit/prediction.py b/src/skit/prediction.py
@@ -198,7 +198,8 @@ def tf_wrong_predictions(dataset, model, labels, qt_desired, with_x_test=False,
             if incorrect_count >= qt_desired:
                 break
 
-        # Close the progress bar when done
+        # Show and close the progress bar when done
+        pbar.update(qt_desired)
         pbar.close()
 
         if with_x_test: