From fd4b04d186dc494740a025932dad08d511af6684 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Wed, 28 Jul 2021 10:56:26 +0300
Subject: [PATCH] [~] Refactor

---
 python/tasks/mlb_player.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index 463d699..a6c08f1 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -1697,7 +1697,7 @@ def kernel_27():
             p.wait()
             assert p.returncode == 0
 
-        assert os.system(r'''
+        with subprocess.Popen('''
             cd /kaggle/working/AlphaPose && \
             python3 \
                 scripts/demo_inference.py \
@@ -1705,4 +1705,7 @@ def kernel_27():
                 --checkpoint pretrained_models/fast_res50_256x192.pth \
                 --video %s \
                 --outdir %s
-        ''' % (t4, t2)) == 0
+        ''' % (t4, t2), stdout=subprocess.PIPE, stderr=subprocess.PIPE)  as p:
+            pprint.pprint(p.communicate())
+            p.wait()
+            assert p.returncode == 0