From 235a050622256627ad3f0ef856cbb12d80e2c78e Mon Sep 17 00:00:00 2001
From: GuoYu Yang <49673553+pilgrimygy@users.noreply.github.com>
Date: Thu, 15 Apr 2021 10:01:55 +0800
Subject: [PATCH] Fix a bug

---
 src/experiments/gridworlds/JuliaRL_BasicDQN_EmptyRoom.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/experiments/gridworlds/JuliaRL_BasicDQN_EmptyRoom.jl b/src/experiments/gridworlds/JuliaRL_BasicDQN_EmptyRoom.jl
index 3d1c8b2..33ba722 100644
--- a/src/experiments/gridworlds/JuliaRL_BasicDQN_EmptyRoom.jl
+++ b/src/experiments/gridworlds/JuliaRL_BasicDQN_EmptyRoom.jl
@@ -14,7 +14,7 @@ function RLCore.Experiment(
     lg = TBLogger(log_dir, min_level = Logging.Info)
     rng = StableRNG(seed)
 
-    inner_env = GridWorlds.EmptyRoom(rng = rng)
+    inner_env = GridWorlds.EmptyRoomDirected(rng = rng)
     action_space_mapping = x -> Base.OneTo(length(RLBase.action_space(inner_env)))
     action_mapping = i -> RLBase.action_space(inner_env)[i]
     env = RLEnvs.ActionTransformedEnv(