From a2d6d514d55d1c64f028817dadbe1402b3a16884 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 7 Aug 2019 10:19:03 -0400
Subject: [PATCH] cleaned up pep8 issues

---
 pytorch_lightning/models/trainer.py | 2 +-
 tests/test_models.py                | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 1b3a9a7a..2dbf46eb 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -261,7 +261,7 @@ class Trainer(TrainerIO):
 
             if '.ckpt' in name:
                 epoch = name.split('epoch_')[1]
-                epoch = int(re.sub('[^0-9]', '' ,epoch))
+                epoch = int(re.sub('[^0-9]', '', epoch))
 
                 if epoch > last_epoch:
                     last_epoch = epoch
diff --git a/tests/test_models.py b/tests/test_models.py
index 746f97ef..f6e792fe 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -123,8 +123,6 @@ def test_amp_gpu_ddp():
     run_gpu_model_test(trainer_options, model, hparams)
 
 
-
-
 def test_cpu_slurm_save_load():
     """
     Verify model save/load/checkpoint on CPU