env var cleanu

Author: ved1beta

tests/tests_pytorch/conftest.py

@@ -95,6 +95,15 @@ def restore_env_variables():
         "TF_GRPC_DEFAULT_OPTIONS",
         "XLA_FLAGS",
         "TORCHINDUCTOR_CACHE_DIR",  # leaked by torch.compile
+        # Memory leak test related
+        "PYTORCH_CUDA_ALLOC_CONF",  # PyTorch memory allocator config
+        "CUDA_VISIBLE_DEVICES",  # GPU visibility
+        "PYTORCH_NO_CUDA_MEMORY_CACHING",  # Disable CUDA memory caching
+        # TensorFlow and TPU related
+        "ENABLE_RUNTIME_UPTIME_TELEMETRY",  # TensorFlow telemetry
+        "TF2_BEHAVIOR",  # TensorFlow 2.x behavior flag
+        "TPU_ML_PLATFORM",  # TPU platform configuration
+        "TPU_ML_PLATFORM_VERSION",  # TPU platform version
     }
     leaked_vars.difference_update(allowlist)
     assert not leaked_vars, f"test is leaking environment variable(s): {set(leaked_vars)}"