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