5a60ab34a4cecfe0b9636f6df77c087faa8b6959,tests/tracking/test_fluent.py,,test_search_runs_attributes,#,359

Before Change

def test_search_runs_attributes():
    runs = [create_run(status=RunStatus.FINISHED, a_uri="dbfs:/test", run_id="abc", exp_id="123"),
            create_run(status=RunStatus.SCHEDULED, a_uri="dbfs:/test2", run_id="def", exp_id="321")]
    with mock.patch.object(MlflowClient, "search_runs", return_value=runs):
        pdf = search_runs()
        data = {"status": [RunStatus.FINISHED, RunStatus.SCHEDULED],
                "artifact_uri": ["dbfs:/test", "dbfs:/test2"],

After Change

def test_search_runs_attributes():
    runs = [create_run(status=RunStatus.FINISHED, a_uri="dbfs:/test", run_id="abc", exp_id="123"),
            create_run(status=RunStatus.SCHEDULED, a_uri="dbfs:/test2", run_id="def", exp_id="321")]
    with mock.patch("mlflow.tracking.fluent._get_paginated_runs", return_value=runs):
        pdf = search_runs()
        data = {"status": [RunStatus.FINISHED, RunStatus.SCHEDULED],
                "artifact_uri": ["dbfs:/test", "dbfs:/test2"],

Instances