From 24ceafa05cf9660b68d4e3e1d24dfccc0f5f934e Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 12:14:26 -0400
Subject: [PATCH] refactored model tests

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

diff --git a/tests/test_models.py b/tests/test_models.py
index 7bf4ee3f..a39876b6 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -184,8 +184,8 @@ def run_gpu_model_test(trainer_options):
     checkpoint = ModelCheckpoint(save_dir)
 
     # add these to the trainer options
-    trainer_options.checkpoint_callback = checkpoint
-    trainer_options.experiment = exp
+    trainer_options['checkpoint_callback'] = checkpoint
+    trainer_options['experiment'] = exp
 
     # fit model
     trainer = Trainer(**trainer_options)