fix sac eval mode (#169)

* fix sac eval mode * update continuous test to use LunarLander
cpnota · Sep 29, 2020 · 72a5ffc · 72a5ffc
1 parent 867ba55
commit 72a5ffc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/all/agents/sac.py b/all/agents/sac.py
@@ -71,7 +71,7 @@ def act(self, state):
         return self._action
 
     def eval(self, state):
-        return self.policy.eval(state)[0]
+        return self.policy.eval(state)
 
     def _train(self):
         if self._should_train():

diff --git a/all/presets/continuous_test.py b/all/presets/continuous_test.py
@@ -14,7 +14,7 @@ def test_sac(self):
         self.validate(sac(replay_start_size=50, device='cpu'))
 
     def validate(self, make_agent):
-        validate_agent(make_agent, GymEnvironment('Pendulum-v0'))
+        validate_agent(make_agent, GymEnvironment('LunarLanderContinuous-v2'))
 
 if __name__ == '__main__':
     unittest.main()