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