Skip to content

Commit

Permalink
more visualization
Browse files Browse the repository at this point in the history
Signed-off-by: Yicheng-Lu-llll <luyc58576@gmail.com>
  • Loading branch information
Yicheng-Lu-llll committed Jun 7, 2023
1 parent 6034708 commit a0d1f95
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions flytekit/bin/entrypoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,6 @@ def _dispatch_execute(
try:
# Step1
local_inputs_file = os.path.join(ctx.execution_state.working_dir, "inputs.pb")
print("inputs_path: ", inputs_path)
ctx.file_access.get_data(inputs_path, local_inputs_file)
input_proto = utils.load_proto_from_file(_literals_pb2.LiteralMap, local_inputs_file)
idl_input_literals = _literal_models.LiteralMap.from_flyte_idl(input_proto)
Expand Down Expand Up @@ -160,8 +159,8 @@ def _dispatch_execute(

ctx.file_access.put_data(ctx.execution_state.engine_dir, output_prefix, is_multipart=True)
_output_deck(task_def.name.split(".")[-1], ctx.user_space_params)

logger.info(f"Engine folder written successfully to the output prefix {output_prefix}")
logger.debug("Finished _dispatch_execute")

if os.environ.get("FLYTE_FAIL_ON_ERROR", "").lower() == "true" and _constants.ERROR_FILE_NAME in output_file_dict:
# This env is set by the flytepropeller
Expand Down

0 comments on commit a0d1f95

Please sign in to comment.