def test_setup_with_invalid_params(self):
hparams = HParams()
hparams.memory_size = 100
# memory_priority_control and memory_priority_compensation should be <= 1
hparams.memory_priority_control = 1.75
hparams.memory_priority_compensation = 0.8
with self.assertRaises(AssertionError):
memory = PrioritizedMemory(hparams, 0)
hparams.memory_priority_control = 1.75
hparams.memory_priority_compensation = 1.8
with self.assertRaises(AssertionError):
memory = PrioritizedMemory(hparams, 0)
def get_empty_memory(self):
hparams = HParams()
hparams.memory_size = 100
return SimpleMemory(hparams, 0)
def get_empty_memory(self): hparams = HParams() hparams.memory_size = 100 hparams.memory_priority_control = 1 hparams.memory_priority_compensation = 1 return PrioritizedMemory(hparams, 0)