Env--

ilabcode · Jan 29, 2024 · 2366d08 · 2366d08
1 parent 65cddab
commit 2366d08
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/environment.jl b/src/environment.jl
@@ -54,6 +54,7 @@ function step!(env::EpistChainEnv, action_label::String)
 
     # Needed for reward condition indexing
     reward_conditions = ["TOP", "BOTTOM"]
+    reward_locations = [(2,6), (4,6)]
     rew_cond_idx = Dict(reward_conditions[1] => 1, reward_conditions[2] => 2)
     #------------------------------------