From c61e13f0ffa13d11b66a98538bf74ddbf655b179 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Fri, 26 Jul 2019 22:13:41 -0400
Subject: [PATCH] fixed hpc save, load. cleaned apu

---
 tests/test_models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_models.py b/tests/test_models.py
index bdd31a5c..5ead6de8 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -66,6 +66,10 @@ def test_dp_output_reduce():
 
 
 def test_model_saving_loading():
+    """
+    Tests use case where trainer saves the model, and user loads it from tags independently
+    :return:
+    """
     hparams = get_hparams()
     model = LightningTestModel(hparams)