0d3690f88f1a03b738953227b2d00a9bad6d7be9,tests/tracking/test_tracking.py,,test_set_tags,#Any#,225

Before Change

def test_set_tags(tracking_uri_mock):
    exact_expected_tags = {"name_1": "c", "name_2": "b", "nested/nested/name": "5"}
    approx_expected_tags = set([MLFLOW_SOURCE_NAME, MLFLOW_SOURCE_TYPE])
    active_run = start_run()
    run_uuid = active_run.info.run_uuid
    with active_run:
        mlflow.set_tags(exact_expected_tags)
    finished_run = tracking.MlflowClient().get_run(run_uuid)
    // Validate tags
    assert len(finished_run.data.tags) == len(exact_expected_tags) + len(approx_expected_tags)
    for tag_key, tag_val in finished_run.data.tags.items():

After Change

def test_set_tags(tracking_uri_mock):
    exact_expected_tags = {"name_1": "c", "name_2": "b", "nested/nested/name": "5"}
    approx_expected_tags = set([MLFLOW_SOURCE_NAME, MLFLOW_SOURCE_TYPE])
    with start_run() as active_run:
        run_uuid = active_run.info.run_uuid
        mlflow.set_tags(exact_expected_tags)
    finished_run = tracking.MlflowClient().get_run(run_uuid)
    // Validate tags
    assert len(finished_run.data.tags) == len(exact_expected_tags) + len(approx_expected_tags)
    for tag_key, tag_val in finished_run.data.tags.items():

Instances