format

Lightning-AI · Jul 23, 2020 · b0860b0 · b0860b0
1 parent f336507
commit b0860b0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pytorch_lightning/trainer/distrib_data_parallel.py b/pytorch_lightning/trainer/distrib_data_parallel.py
@@ -162,9 +162,9 @@ def train_fx(trial_hparams, cluster_manager, _):
 else:
     XLA_AVAILABLE = True
 
-pid = os.getpid()
-rng1 = np.random.RandomState(pid)
-RANDOM_PORTS = rng1.randint(10000, 19999, 100)
+PID = os.getpid()
+RNG1 = np.random.RandomState(PID)
+RANDOM_PORTS = RNG1.randint(10000, 19999, 100)
 
 
 class TrainerDDPMixin(ABC):
@@ -345,7 +345,6 @@ def configure_slurm_ddp(self, num_gpu_nodes):
     def determine_local_rank(self):
         if self.is_slurm_managing_tasks:
             return int(os.environ['SLURM_LOCALID'])
-
         else:
             return int(os.environ.get('LOCAL_RANK', 0))