This commit is contained in:
Trevor Blackwell
2017-03-15 11:54:44 -07:00
parent 2a0a2a3a2b
commit cb8ca5a4aa

View File

@@ -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