Prevent clean_up_envs from deleting local development environment (m…

…lflow#11249)

Signed-off-by: harupy <[email protected]>

conftest.py

@@ -1,6 +1,7 @@
 import json
 import os
 import posixpath
+import re
 import shutil
 import subprocess
 import sys
@@ -263,6 +264,9 @@ def pytest_terminal_summary(terminalreporter, exitstatus, config):
 
 @pytest.fixture(scope="module", autouse=True)
 def clean_up_envs():
+    """
+    Clean up virtualenvs and conda environments created during tests to save disk space.
+    """
     yield
 
     if "GITHUB_ACTIONS" in os.environ:
@@ -272,8 +276,11 @@ def clean_up_envs():
         if os.name != "nt":
             conda_info = json.loads(subprocess.check_output(["conda", "info", "--json"], text=True))
             root_prefix = conda_info["root_prefix"]
+            regex = re.compile(r"mlflow-\w{32,}")
             for env in conda_info["envs"]:
-                if env != root_prefix:
+                if env == root_prefix:
+                    continue
+                if regex.fullmatch(os.path.basename(env)):
                     shutil.rmtree(env, ignore_errors=True)