use monkeypatch context

Author: fnhirwa

tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py

@@ -493,12 +493,13 @@ def test_strategy_choice_ddp_torchelastic(_, __, mps_count_0, cuda_count_2):
 )
 def test_torchelastic_priority_over_slurm(monkeypatch):
     """Test that the TorchElastic cluster environment is chosen over SLURM when both are detected."""
-    mock_cuda_count(monkeypatch, 2)
-    mock_mps_count(monkeypatch, 0)
-    mock_hpu_count(monkeypatch, 0)
+    with monkeypatch.context():
+        mock_cuda_count(monkeypatch, 2)
+        mock_mps_count(monkeypatch, 0)
+        mock_hpu_count(monkeypatch, 0)
+        connector = _AcceleratorConnector(strategy="ddp")
     assert TorchElasticEnvironment.detect()
     assert SLURMEnvironment.detect()
-    connector = _AcceleratorConnector(strategy="ddp")
     assert isinstance(connector.strategy.cluster_environment, TorchElasticEnvironment)