Update slimevolley.py

Ensures otherAction is converted when using atari mode and a multi-agent setup
frsrgrrw · May 10, 2021 · 5bf9186 · 5bf9186
1 parent e840033
commit 5bf9186
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/slimevolleygym/slimevolley.py b/slimevolleygym/slimevolley.py
@@ -775,15 +775,15 @@ def step(self, action, otherAction=None):
 
     if self.otherAction is not None:
       otherAction = self.otherAction
+
+    if otherAction is None: # override baseline policy
+      obs = self.game.agent_left.getObservation()
+      otherAction = self.policy.predict(obs)
 
     if self.atari_mode:
       action = self.discreteToBox(action)
-      if otherAction is not None:
-        otherAction = self.discreteToBox(otherAction)
+      otherAction = self.discreteToBox(otherAction)
 
-    if otherAction is None: # override baseline policy
-      obs = self.game.agent_left.getObservation()
-      otherAction = self.policy.predict(obs)
     self.game.agent_left.setAction(otherAction)
     self.game.agent_right.setAction(action) # external agent is agent_right