Apply suggestions from code review

Co-authored-by: Carlos Mocholí <[email protected]>

Author: four4fish

pytorch_lightning/strategies/parallel.py

@@ -85,14 +85,6 @@ def distributed_sampler_kwargs(self):
         distributed_sampler_kwargs = dict(num_replicas=len(self.parallel_devices), rank=self.global_rank)
         return distributed_sampler_kwargs
 
-    @property
-    def parallel_devices(self):
-        return self._parallel_devices
-
-    @parallel_devices.setter
-    def parallel_devices(self, parallel_devices):
-        self._parallel_devices = parallel_devices
-
     def reconciliate_processes(self, trace: str):
         """Function to re-conciliate processes on failure."""
 

pytorch_lightning/utilities/device_parser.py

@@ -21,7 +21,7 @@
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
 
-def determine_root_gpu_device(gpus: List[Union[int, torch.device]]) -> Optional[Union[int, torch.device]]:
+def determine_root_gpu_device(gpus: List[_DEVICE]) -> Optional[_DEVICE]:
     """
     Args:
         gpus: non-empty list of ints representing which gpus to use