From 322436519096b65585424098683517bd2fdc3035 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 26 Jul 2019 22:39:44 -0400
Subject: [PATCH] added clean slurm save load test

---
 tests/test_models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_models.py b/tests/test_models.py
index c83ff428e8..57fe9f7702 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -74,8 +74,9 @@ def test_cpu_slurm_save_load():
     saved_filepath = trainer.hpc_save(save_dir, exp)
     assert os.path.exists(saved_filepath)
 
-    # wipe-out trainer model
+    # wipe-out trainer and model
     # we want to see if the weights come back correctly
+    trainer = Trainer(**trainer_options)
     trainer.model = LightningTestModel(hparams)
 
     # test HPC loading