From 4ffd5e17d830784274ee003cce18526b4db13a66 Mon Sep 17 00:00:00 2001 From: Min Shi Date: Mon, 23 Oct 2023 17:55:58 +0800 Subject: [PATCH] fix flake --- .../promptflow/_sdk/operations/_test_submitter.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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,