diff --git a/src/promptflow/promptflow/_sdk/operations/_test_submitter.py b/src/promptflow/promptflow/_sdk/operations/_test_submitter.py index 4137d53e02f5..3cd285e2e58c 100644 --- a/src/promptflow/promptflow/_sdk/operations/_test_submitter.py +++ b/src/promptflow/promptflow/_sdk/operations/_test_submitter.py @@ -157,7 +157,8 @@ def flow_test( environment_variables = environment_variables if environment_variables else {} SubmitterHelper.init_env(environment_variables=environment_variables) - with LoggerOperations(file_path=self.flow.code / PROMPT_FLOW_DIR_NAME / "flow.log", stream=stream_log, credential_list=credential_list): + with LoggerOperations(file_path=self.flow.code / PROMPT_FLOW_DIR_NAME / "flow.log", \ + stream=stream_log, credential_list=credential_list): storage = DefaultRunStorage(base_dir=self.flow.code, sub_dir=Path(".promptflow/intermediate")) flow_executor = FlowExecutor.create( self.flow.path, connections, self.flow.code, storage=storage, raise_ex=False @@ -199,7 +200,8 @@ def node_test( SubmitterHelper.resolve_environment_variables(environment_variables=environment_variables, client=self._client) SubmitterHelper.init_env(environment_variables=environment_variables) - with LoggerOperations(file_path=self.flow.code / PROMPT_FLOW_DIR_NAME / f"{node_name}.node.log", stream=stream, credential_list=credential_list): + with LoggerOperations(file_path=self.flow.code / PROMPT_FLOW_DIR_NAME / f"{node_name}.node.log", \ + stream=stream, credential_list=credential_list): result = FlowExecutor.load_and_exec_node( self.flow.path, node_name,