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

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

diff --git a/tests/test_models.py b/tests/test_models.py
index 57fe9f7702..2977f46073 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -80,9 +80,8 @@ def test_cpu_slurm_save_load():
     trainer.model = LightningTestModel(hparams)
 
     # test HPC loading
-    trainer.global_step = 20000000
     trainer.hpc_load(save_dir, on_gpu=False)
-    assert trainer.global_step == real_global_step and trainer.global_step != 20000000
+    assert trainer.global_step == real_global_step and trainer.global_step > 0
 
     # predict with loaded model to make sure answers are the same
     trainer.model.eval()