Update test_pipeline_run.py

lu-ohai · web-flow · commit 9169b21cab0c · 2024-07-18T13:08:33.000-04:00
diff --git a/tests/unitary/default_setup/pipeline/test_pipeline_run.py b/tests/unitary/default_setup/pipeline/test_pipeline_run.py
@@ -391,6 +391,8 @@ def test_stream_log(self, mock_stream, mock_sync):
             **PIPELINE_RUN_LOG_DETAILS
         )
         pipeline_run.time_accepted = datetime.now()
+        service_logging = OCILog()
+        pipeline_run._set_service_logging_resource(service_logging)
         pipeline_run._PipelineRun__stream_log(
             ConsolidatedLog(OCILog()),
             [custom_script_step.step_name, ml_job_step.step_name],

Original file line number	Diff line number	Diff line change
`@@ -391,6 +391,8 @@ def test_stream_log(self, mock_stream, mock_sync):`
`391`	`391`	`**PIPELINE_RUN_LOG_DETAILS`
`392`	`392`	`)`
`393`	`393`	`pipeline_run.time_accepted = datetime.now()`
	`394`	`+ service_logging = OCILog()`
	`395`	`+ pipeline_run._set_service_logging_resource(service_logging)`
`394`	`396`	`pipeline_run._PipelineRun__stream_log(`
`395`	`397`	`ConsolidatedLog(OCILog()),`
`396`	`398`	`[custom_script_step.step_name, ml_job_step.step_name],`