update

microsoft · Jan 7, 2025 · e6d80c5 · e6d80c5
1 parent d352938
commit e6d80c5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/promptflow-executor-e2e-test.yml b/.github/workflows/promptflow-executor-e2e-test.yml
@@ -27,7 +27,6 @@ on:
       - .github/workflows/promptflow-executor-e2e-test.yml
   workflow_dispatch:
 env:
-  PF_DISABLE_TRACING: "false"
   packageSetupType: promptflow_with_extra
   testWorkingDirectory: ${{ github.workspace }}/src/promptflow
   PYTHONPATH: ${{ github.workspace }}/src/promptflow

diff --git a/.github/workflows/promptflow-executor-unit-test.yml b/.github/workflows/promptflow-executor-unit-test.yml
@@ -31,7 +31,6 @@ permissions:
   id-token: write
   contents: read
 env:
-  PF_DISABLE_TRACING: "false"
   packageSetupType: promptflow_with_extra
   testWorkingDirectory: ${{ github.workspace }}/src/promptflow
   PYTHONPATH: ${{ github.workspace }}/src/promptflow

diff --git a/src/promptflow/tests/executor/conftest.py b/src/promptflow/tests/executor/conftest.py
@@ -1,4 +1,5 @@
 import multiprocessing
+import os
 from pathlib import Path
 
 import pytest
@@ -39,6 +40,9 @@ def is_replay():
         return False
 
 
+# Enable tracing for testing
+os.environ["PF_DISABLE_TRACING"] = False
+
 PROMPTFLOW_ROOT = Path(__file__) / "../../.."