stupid formatting

Lightning-AI · Feb 1, 2021 · ea88661 · ea88661
1 parent 3a44c73
commit ea88661
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 12 deletions.
diff --git a/pytorch_lightning/accelerators/accelerator_connector.py b/pytorch_lightning/accelerators/accelerator_connector.py
@@ -90,7 +90,7 @@ def __init__(
         self.replace_sampler_ddp = replace_sampler_ddp
         self.deterministic = deterministic
         self.precision = precision
-        self.amp_type =  amp_type.lower() if isinstance(amp_type, str) else None
+        self.amp_type = amp_type.lower() if isinstance(amp_type, str) else None
         self.amp_level = amp_level
         self.cluster_environment = cluster_environment
         self.is_slurm_managing_tasks = False

diff --git a/pytorch_lightning/plugins/training_type/tpu_spawn.py b/pytorch_lightning/plugins/training_type/tpu_spawn.py
@@ -182,18 +182,12 @@ def __load_weights_on_main_process(self) -> None:
     def xmp_spawn_kwargs(self):
         return {
             "args": (self.lightning_module, trainer, self.mp_queue),
-             "nproc": len(self.parallel_devices),
-             "start_method": self.start_method
-          }
+            "nproc": len(self.parallel_devices),
+            "start_method": self.start_method
+        }
 
     def start_training(self, trainer) -> None:
-        xmp.spawn(
-            self.new_process,
-            **self.xmp_spawn_kwargs
-        )
+        xmp.spawn(self.new_process, **self.xmp_spawn_kwargs)
 
     def start_testing(self, trainer) -> None:
-        xmp.spawn(
-            self.new_process,
-            **self.xmp_spawn_kwargs
-        )
+        xmp.spawn(self.new_process, **self.xmp_spawn_kwargs)