From 790fde2bb1946c6df005c8b5059b31c56c8b27be Mon Sep 17 00:00:00 2001 From: Yee Hing Tong Date: Tue, 9 Jul 2024 23:08:01 -0700 Subject: [PATCH] remove prints Signed-off-by: Yee Hing Tong --- plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py index 94d03a6887..6b7d8e76b2 100644 --- a/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py +++ b/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py @@ -256,11 +256,9 @@ def spawn_helper( try: return_val = fn(**kwargs) omt = ctx.output_metadata_tracker - logger.warning(f"ARTFELASTIC: 1 {omt}") om = None if omt: om = omt.get(return_val) - logger.warning(f"ARTFELASTIC: 2 {om=}") except Exception as e: # See explanation in `create_recoverable_error_file` why we check # for recoverable errors here in the worker processes. @@ -445,9 +443,7 @@ def fn_partial(): for deck in out[0].decks: if not isinstance(deck, flytekit.deck.deck.TimeLineDeck): ctx.decks.append(deck) - logger.warning(f"ARTFELASTIC: 3 {out[0]}") if out[0].om: - logger.warning(f"ARTFELASTIC: 4 Adding back {out[0].om}") ctx.output_metadata_tracker.add(out[0].return_value, out[0].om) return out[0].return_value