Replace set by system environment variable with class attribute

src/promptflow-evals/promptflow/evals/_constants.py

@@ -25,4 +25,3 @@ class Prefixes:
 
 PF_BATCH_TIMEOUT_SEC_DEFAULT = 3600
 PF_BATCH_TIMEOUT_SEC = "PF_BATCH_TIMEOUT_SEC"
-PF_IS_BATCH_TIMEOUT_SYSTEM_SET = "PF_IS_BATCH_TIMEOUT_SYSTEM_SET"

src/promptflow-evals/promptflow/evals/evaluate/_batch_run_client/batch_run_context.py

@@ -5,11 +5,7 @@
 
 from promptflow._sdk._constants import PF_FLOW_ENTRY_IN_TMP, PF_FLOW_META_LOAD_IN_SUBPROCESS
 from promptflow._utils.user_agent_utils import ClientUserAgentUtil
-from promptflow.evals._constants import (
-    PF_BATCH_TIMEOUT_SEC,
-    PF_BATCH_TIMEOUT_SEC_DEFAULT,
-    PF_IS_BATCH_TIMEOUT_SYSTEM_SET,
-)
+from promptflow.evals._constants import PF_BATCH_TIMEOUT_SEC, PF_BATCH_TIMEOUT_SEC_DEFAULT
 from promptflow.tracing._integrations._openai_injector import inject_openai_api, recover_openai_api
 
 from ..._user_agent import USER_AGENT
@@ -21,6 +17,7 @@
 class BatchRunContext:
     def __init__(self, client):
         self.client = client
+        self._is_timeout_set_by_system = False
 
     def __enter__(self):
         if isinstance(self.client, CodeClient):
@@ -33,9 +30,7 @@ def __enter__(self):
 
             if os.environ.get(PF_BATCH_TIMEOUT_SEC) is None:
                 os.environ[PF_BATCH_TIMEOUT_SEC] = str(PF_BATCH_TIMEOUT_SEC_DEFAULT)
-                os.environ[PF_IS_BATCH_TIMEOUT_SYSTEM_SET] = "true"
-            else:
-                os.environ[PF_IS_BATCH_TIMEOUT_SYSTEM_SET] = "false"
+                self._is_timeout_set_by_system = True
 
             # For addressing the issue of asyncio event loop closed on Windows
             set_event_loop_policy()
@@ -48,6 +43,6 @@ def __exit__(self, exc_type, exc_val, exc_tb):
             os.environ.pop(PF_FLOW_ENTRY_IN_TMP, None)
             os.environ.pop(PF_FLOW_META_LOAD_IN_SUBPROCESS, None)
 
-            if os.environ.get(PF_IS_BATCH_TIMEOUT_SYSTEM_SET) == "true":
+            if self._is_timeout_set_by_system:
                 os.environ.pop(PF_BATCH_TIMEOUT_SEC, None)
-                os.environ[PF_IS_BATCH_TIMEOUT_SYSTEM_SET] = "false"
+                self._is_timeout_set_by_system = False