From bf3b86ce4d6597e885f3fe980d616b8fd86bcc5b Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Tue, 25 Jun 2019 18:58:45 -0400
Subject: [PATCH] updated args

---
 docs/source/examples/fully_featured_trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/examples/fully_featured_trainer.py b/docs/source/examples/fully_featured_trainer.py
index ac38ed18ea..e000885d60 100644
--- a/docs/source/examples/fully_featured_trainer.py
+++ b/docs/source/examples/fully_featured_trainer.py
@@ -95,7 +95,7 @@ def main(hparams, cluster, results_dict):
     # gpus are ; separated for inside a node and , within nodes
     gpu_list = None
     if hparams.gpus is not None:
-        gpu_list = map(int, hparams.gpus.split(';'))
+        gpu_list = [int(x) for x in hparams.gpus.split(';')]
 
     # configure trainer
     trainer = Trainer(