def setup_method(self):
self.network = DummyNetwork()
self.rollout = Rollout()
self.metrics = DummyMetrics()
self.controller = PPOController(
self.network, self.rollout, self.metrics, num_envs=4,
time_horizon=128, epoch=4, batch_size=32, gamma=0.99, lam=0.9)
def setUp(self):
self.network = DummyNetwork()
self.buffer = Buffer()
self.metrics = DummyMetrics()
self.noise = DummyNoise()
self.controller = DDPGController(self.network,
self.buffer,
self.metrics,
self.noise,
num_actions=4,
batch_size=32)
def setup_method(self):
self.network = DummyNetwork()
self.metrics = DummyMetrics()
self.metrics.has = MagicMock(return_value=True)
self.controller = EvalController(self.network, self.metrics, 10)