From 2fa756018f9f4d91c5c6040dc13544e30aa635e8 Mon Sep 17 00:00:00 2001
From: Chuck Tang <chuck@mosaicml.com>
Date: Mon, 2 Oct 2023 11:41:32 -0700
Subject: [PATCH] add

---
 .github/mcp/mcp_pytest.py         | 3 ++-
 .github/workflows/pytest-cpu.yaml | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/mcp/mcp_pytest.py b/.github/mcp/mcp_pytest.py
index 4c297088a1..94f6f644b0 100644
--- a/.github/mcp/mcp_pytest.py
+++ b/.github/mcp/mcp_pytest.py
@@ -86,11 +86,12 @@
     if len(name) > 56:
         name = name[:56]
 
+    clear_tmp_path_flag = '-o tmp_path_retention_policy=none'
     command += f'''
 
     pip install --upgrade --user .[all]
 
-    export COMMON_ARGS="-v --durations=20 -m '{args.pytest_markers}'"
+    export COMMON_ARGS="-v --durations=20 -m '{args.pytest_markers}' {clear_tmp_path_flag}"
 
     make test PYTEST='{args.pytest_command}' EXTRA_ARGS="$COMMON_ARGS --codeblocks"
 
diff --git a/.github/workflows/pytest-cpu.yaml b/.github/workflows/pytest-cpu.yaml
index d66d18b5df..3bb3697b91 100644
--- a/.github/workflows/pytest-cpu.yaml
+++ b/.github/workflows/pytest-cpu.yaml
@@ -33,7 +33,7 @@ jobs:
       run: |
         set -ex
         export PATH=/composer-python:$PATH
-        export COMMON_ARGS="-v --durations=20 -m '${{ inputs.pytest-markers }}'"
+        export COMMON_ARGS="-v --durations=20 -m '${{ inputs.pytest-markers }}' -o tmp_path_retention_policy=none"
 
         # Necessary to run git diff for doctests
         git config --global --add safe.directory /__w/llm-foundry/llm-foundry