From 80bcbee4a537119b047001e63a8d20a260daa555 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Tue, 20 Jul 2021 10:05:31 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/mlb_player.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index 4e0b20e..2a75e3b 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -241,6 +241,7 @@ def kernel_6(
     if max_frames is None:
         max_frames = 10
 
+    import tqdm
     import cv2
 
     t1 = glob.glob('*.mp4')
@@ -251,7 +252,7 @@ def kernel_6(
         t2 = None
         try:
             t2 = cv2.VideoCapture(o)
-            for k in range(10):
+            for k in tqdm.tqdm(range(max_frames)):
                 t3 = t2.read()
                 assert t3[0]
                 t4 = t3[1]