class Node(object): """Generic Node class. Used in the implementation of a prioritised replay buffer""" def __init__(self, key, value): self.key = key self.value = value def update_key_and_value(self, new_key, new_value): self.update_key(new_key) self.update_value(new_value) def update_key(self, new_key): self.key = new_key def update_value(self, new_value): self.value = new_value def __eq__(self, other): return self.key == other.key and self.value == other.value