cf05c5bd1943dd0f855c3d2a51c904e775e6eae7,tests/tracking/test_fluent.py,,test_start_run_with_parent,#,354

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()

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

    with databricks_notebook_patch, active_run_stack_patch, \
            mock.patch.object(MlflowClient, "create_run"), \
            mock.patch.object(SourceType, "to_string", return_value="type_1248"), \
            mock.patch.object(SourceType, "from_string", return_value=mock_source_type):
        active_run = start_run(
            experiment_id=mock_experiment_id, source_name=mock_source_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,
        mlflow_tags.MLFLOW_PARENT_RUN_ID: parent_run.info.run_uuid

Instances