Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds neptune plugin for experiment tracking #2686

Merged
merged 13 commits into from
Aug 26, 2024
Prev Previous commit
Next Next commit
Fix tests with new names
Signed-off-by: Thomas J. Fan <thomasjpfan@gmail.com>
thomasjpfan committed Aug 21, 2024

Verified

This commit was signed with the committer’s verified signature.
thomasjpfan Thomas J. Fan
commit 5727be2be7a08b06c5a6db779f03a3fcddbe8e8b
4 changes: 2 additions & 2 deletions plugins/flytekit-neptune/tests/test_neptune_init_run.py
Original file line number Diff line number Diff line change
@@ -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():