Fix tests with new names

Signed-off-by: Thomas J. Fan <[email protected]>
flyteorg · Aug 21, 2024 · 5727be2 · 5727be2
1 parent c4a06da
commit 5727be2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plugins/flytekit-neptune/tests/test_neptune_init_run.py b/plugins/flytekit-neptune/tests/test_neptune_init_run.py
@@ -70,8 +70,8 @@ def test_remote_project_and_init_run_kwargs(
     neptune_task()
 
     init_run_mock.assert_called_with(project="flytekit/project", tags=["my-tag"], api_token=api_token)
-    assert run_mock["Flyte Execution ID"] == host_name
-    assert run_mock["Flyte Execution URL"] == execution_url
+    assert run_mock["flyte/execution_id"] == host_name
+    assert run_mock["flyte/execution_url"] == execution_url
 
 
 def test_get_secret_callable():