Skip to content

Commit

Permalink
Use flyte namespace for logging
Browse files Browse the repository at this point in the history
Signed-off-by: Thomas J. Fan <[email protected]>
  • Loading branch information
thomasjpfan committed Aug 21, 2024
1 parent e36c2c6 commit 620cde5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions plugins/flytekit-neptune/flytekitplugins/neptune/tracking.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 620cde5

Please sign in to comment.