def blank_brain():
config = Config([__name__])
return Brain(config)
def record_csv_config():
return Config([
__name__, '--accesskey=VALUE', '--username=alice',
'--url=http://localhost:8889', '--brain=cartpole', '--proxy=VALUE',
'--record=foobar.csv'
])
self._computer_turn()
return self.render(), self.reward, self.terminal
class TicTacToeSim(Simulator):
def __init__(self, brain, iteration_limit=0):
super(TicTacToeSim, self).__init__(brain, 'tictactoe_simulator')
self._env = TicTacToeGame()
def episode_finish(self):
log.gym("Episode {} reward: {}".format(self.episode_count,
self.episode_reward))
def episode_start(self, config):
observation = self._env.reset()
return {'image': observation}
def simulate(self, action):
observation, reward, done = self._env.advance(action)
return {'image': observation}, reward, done
if __name__ == "__main__":
config = Config(sys.argv)
brain = Brain(config)
sim = TicTacToeSim(brain)
while sim.run():
continue
def test_brain_with_proxy():
# Patch and Mock out requests/responses
set_proxies(http_proxy='pass', https_proxy=None, all_proxy=None)
config = Config([__name__])
brain = Brain(config)
assert brain._proxy_header() is not None
def test_brain_no_proxy():
config = Config([__name__])
brain = Brain(config)
assert brain._proxy_header() is None
def test_brain_timeout():
config = Config([__name__, '--network-timeout=1', '--disable-telemetry'])
brain = Brain(config)
assert brain._timeout == 1
def test_brain_no_proxy():
config = Config([__name__, '--disable-telemetry'])
brain = Brain(config)
assert brain._proxy_header() is None
