From 383b4cdac7d8b5ef2bb48dc395829942e244d052 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 16:35:32 -0400
Subject: [PATCH] added sample input for summary

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

diff --git a/tests/test_models.py b/tests/test_models.py
index 2596b303..e31ae60e 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -137,7 +137,7 @@ def test_amp_gpu_dp():
 
     trainer_options = dict(
         max_nb_epochs=1,
-        gpus=[0, 1],
+        gpus='0, 1',  # test init with gpu string
         distributed_backend='dp',
         use_amp=True
     )