def test_start_run_existing_run_from_environment_with_set_environment(empty_active_run_stack): mock_run = mock.Mock() mock_run.info.lifecycle_stage = LifecycleStage.ACTIVE run_id = uuid.uuid4().hex env_patch = mock.patch.dict("os.environ", {_RUN_ID_ENV_VAR: run_id}) with env_patch, mock.patch.object(MlflowClient, "get_run", return_value=mock_run): with pytest.raises(MlflowException): set_experiment("test-run") active_run = start_run()
def test_start_run_existing_run_from_environment_with_set_environment( empty_active_run_stack, ): # pylint: disable=unused-argument mock_run = mock.Mock() mock_run.info.lifecycle_stage = LifecycleStage.ACTIVE run_id = uuid.uuid4().hex env_patch = mock.patch.dict("os.environ", {_RUN_ID_ENV_VAR: run_id}) with env_patch, mock.patch.object(MlflowClient, "get_run", return_value=mock_run): with pytest.raises( MlflowException, match="active run ID does not match environment run ID" ): set_experiment("test-run") start_run()