diff --git a/plugins/flytekit-neptune/flytekitplugins/neptune/tracking.py b/plugins/flytekit-neptune/flytekitplugins/neptune/tracking.py index e23d443665..77527bb5d4 100644 --- a/plugins/flytekit-neptune/flytekitplugins/neptune/tracking.py +++ b/plugins/flytekit-neptune/flytekitplugins/neptune/tracking.py @@ -85,10 +85,10 @@ def execute(self, *args, **kwargs): # The HOSTNAME is set to {.executionName}-{.nodeID}-{.taskRetryAttempt} # If HOSTNAME is not defined, use the execution name as a fallback hostname = os.environ.get("HOSTNAME", ctx.user_space_params.execution_id.name) - run["Flyte Execution ID"] = hostname + run["flyte/execution_id"] = hostname if (execution_url := os.getenv("FLYTE_EXECUTION_URL")) is not None: - run["Flyte Execution URL"] = execution_url + run["flyte/execution_url"] = execution_url new_user_params = ctx.user_space_params.builder().add_attr("NEPTUNE_RUN", run).build() with FlyteContextManager.with_context(