From 710dbf5a12808d639fd5a2a01d9e821f763fd8a0 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Mon, 27 Apr 2020 07:57:09 -0400
Subject: [PATCH] ddp pickle

---
 tests/trainer/test_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
index 9d8217f9..116cafcc 100644
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -30,7 +30,7 @@ from tests.base import TestModelBase
 def test_model_pickle(tmpdir):
     import pickle
 
-    model = TestModelBase()
+    model = TestModelBase(tutils.get_default_hparams())
     pickle.dumps(model)