From cb8ca5a4aa925b834413e30a4fe23cbe8d4c5903 Mon Sep 17 00:00:00 2001 From: Trevor Blackwell Date: Wed, 15 Mar 2017 11:54:44 -0700 Subject: [PATCH] Address https://github.com/openai/gym/issues/523 --- gym/envs/toy_text/frozen_lake.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/gym/envs/toy_text/frozen_lake.py b/gym/envs/toy_text/frozen_lake.py index 490a62fc8..63f450c5a 100644 --- a/gym/envs/toy_text/frozen_lake.py +++ b/gym/envs/toy_text/frozen_lake.py @@ -128,4 +128,5 @@ class FrozenLakeEnv(discrete.DiscreteEnv): outfile.write("\n") outfile.write("\n".join(''.join(line) for line in desc)+"\n") - return outfile + if mode != 'human': + return outfile