cf05c5bd1943dd0f855c3d2a51c904e775e6eae7,tests/tracking/test_fluent.py,,test_start_run_overrides,#Any#,253

Before Change

    mock_experiment_id = mock.Mock()
    mock_source_name = mock.Mock()
    mock_source_type = mock.Mock()
    mock_source_version = mock.Mock()
    mock_entry_point_name = mock.Mock()
    mock_run_name = mock.Mock()

    to_str_patch = mock.patch.object(SourceType, "to_string", return_value="some source type")
    from_str_patch = mock.patch.object(SourceType, "from_string", return_value=mock_source_type)

    expected_tags = {
        mlflow_tags.MLFLOW_SOURCE_NAME: mock_source_name,
        mlflow_tags.MLFLOW_SOURCE_TYPE: "some source type",
        mlflow_tags.MLFLOW_GIT_COMMIT: mock_source_version,
        mlflow_tags.MLFLOW_PROJECT_ENTRY_POINT: mock_entry_point_name
    }

    with databricks_notebook_patch, create_run_patch, to_str_patch, from_str_patch:
        active_run = start_run(
            experiment_id=mock_experiment_id, source_name=mock_source_name,
            source_version=mock_source_version, entry_point_name=mock_entry_point_name,

After Change

    mock_experiment_id = mock.Mock()
    mock_source_name = mock.Mock()
    source_type = SourceType.JOB
    mock_source_version = mock.Mock()
    mock_entry_point_name = mock.Mock()
    mock_run_name = mock.Mock()

    expected_tags = {
        mlflow_tags.MLFLOW_SOURCE_NAME: mock_source_name,
        mlflow_tags.MLFLOW_SOURCE_TYPE: SourceType.to_string(source_type),
        mlflow_tags.MLFLOW_GIT_COMMIT: mock_source_version,
        mlflow_tags.MLFLOW_PROJECT_ENTRY_POINT: mock_entry_point_name
    }

Instances