adjust test

Lightning-AI · Feb 11, 2021 · 9e86ab8 · 9e86ab8
1 parent b0a831a
commit 9e86ab8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/accelerators/legacy/test_accelerator_connector.py b/tests/accelerators/legacy/test_accelerator_connector.py
@@ -287,7 +287,7 @@ def test_accelerator_choice_ddp_cpu_custom_cluster(device_count_mock):
     Test that we choose the custom cluster even when SLURM or TE flags are around
     """
 
-    class CustomCluster(ClusterEnvironment):
+    class CustomCluster(DefaultEnvironment):
 
         def master_address(self):
             return 'asdf'