From b776fce2e7f43ff09d1cbcb6c70b20a6b0fa9e3f Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 08:56:22 -0400
Subject: [PATCH] added test docs

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

diff --git a/tests/test_models.py b/tests/test_models.py
index d34f3b4b..212c3e8f 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -14,6 +14,7 @@ np.random.seed(SEED)
 
 
 def get_model():
+    # set up model with these hyperparams
     root_dir = os.path.dirname(os.path.realpath(__file__))
     hparams = Namespace(**{'drop_prob': 0.2,
                            'batch_size': 32,
@@ -29,6 +30,7 @@ def get_model():
 
 
 def get_exp():
+    # set up exp object without actually saving logs
     exp = Experiment(debug=True)
     return exp
 
@@ -39,6 +41,10 @@ def assert_ok_acc(trainer):
 
 
 def test_cpu_model():
+    """
+    Make sure model trains on CPU
+    :return:
+    """
     model = get_model()
 
     trainer = Trainer(