cleanup artifacts

Author: deependujha

tests/tests_fabric/conftest.py

@@ -132,12 +132,33 @@ def reset_in_fabric_backward():
 @pytest.fixture
 def reset_deterministic_algorithm():
     """Ensures that torch determinism settings are reset before the next test runs."""
-    os.environ["PL_FORCE_DETERMINISTIC_PORTS"] = "1"
     yield
     os.environ.pop("CUBLAS_WORKSPACE_CONFIG", None)
     torch.use_deterministic_algorithms(False)
 
 
+@pytest.fixture(autouse=True)
+def cleanup_lightning_ports():
+    """Delete Lightning deterministic port files after all tests."""
+    from lightning.fabric.plugins.environments.lightning import LOCK_FILE
+
+    # Ensure deterministic ports in tests
+    os.environ["PL_FORCE_DETERMINISTIC_PORTS"] = "1"
+
+    yield
+
+    files_to_remove = [
+        LOCK_FILE,  # the port storage file
+        LOCK_FILE + ".lock",  # the filelock lock file
+    ]
+    for f in files_to_remove:
+        try:
+            if os.path.exists(f):
+                os.remove(f)
+        except Exception as e:
+            print(f"Warning: failed to remove {f}: {e}")
+
+
 @pytest.fixture
 def reset_cudnn_benchmark():
     """Ensures that the `torch.backends.cudnn.benchmark` setting gets reset before the next test runs."""

tests/tests_pytorch/conftest.py

@@ -135,11 +135,32 @@ def teardown_process_group():
 @pytest.fixture(autouse=True)
 def reset_deterministic_algorithm():
     """Ensures that torch determinism settings are reset before the next test runs."""
-    os.environ["PL_FORCE_DETERMINISTIC_PORTS"] = "1"
     yield
     torch.use_deterministic_algorithms(False)
 
 
+@pytest.fixture(autouse=True)
+def cleanup_lightning_ports():
+    """Delete Lightning deterministic port files after all tests."""
+    from lightning.fabric.plugins.environments.lightning import LOCK_FILE
+
+    # Ensure deterministic ports in tests
+    os.environ["PL_FORCE_DETERMINISTIC_PORTS"] = "1"
+
+    yield
+
+    files_to_remove = [
+        LOCK_FILE,  # the port storage file
+        LOCK_FILE + ".lock",  # the filelock lock file
+    ]
+    for f in files_to_remove:
+        try:
+            if os.path.exists(f):
+                os.remove(f)
+        except Exception as e:
+            print(f"Warning: failed to remove {f}: {e}")
+
+
 @pytest.fixture(autouse=True)
 def thread_police_duuu_daaa_duuu_daaa():
     """Attempts to stop left-over threads to avoid test interactions."""