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