old_contents: from __future__ import unicode_literals
  from __future__ import print_function
  from __future__ import division
  from __future__ import absolute_import
  from future import standard_library
  from builtins import *  # NOQA
  standard_library.install_aliases()  # NOQA
  
  import basetest_dqn_like as base
  from chainerrl.agents import SARSA
  
  
  class TestSARSAOnDiscreteABC(base._TestDQNOnDiscreteABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100)
  
  
  class TestSARSAOnContinuousABC(base._TestDQNOnContinuousABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100)
  
  
  class TestSARSAOnDiscretePOABC(base._TestDQNOnDiscretePOABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100,
              episodic_update=True)
  
new_contents: from __future__ import unicode_literals
  from __future__ import print_function
  from __future__ import division
  from __future__ import absolute_import
  from future import standard_library
  from builtins import *  # NOQA
  standard_library.install_aliases()  # NOQA
  
  import basetest_dqn_like as base
  from chainerrl.agents import SARSA
  
  # Sarsa does not support batch training
  
  
  class TestSARSAOnDiscreteABC(base._TestDQNOnDiscreteABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100)
  
  
  class TestSARSAOnContinuousABC(base._TestDQNOnContinuousABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100)
  
  
  class TestSARSAOnDiscretePOABC(base._TestDQNOnDiscretePOABC):
  
      def make_dqn_agent(self, env, q_func, opt, explorer, rbuf, gpu):
          return SARSA(
              q_func, opt, rbuf, gpu=gpu, gamma=0.9, explorer=explorer,
              replay_start_size=100, target_update_interval=100,
              episodic_update=True)
  
subject: Add comment