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(