@override(Policy)
def get_initial_state(self):
    if self.model:
        return self.model.get_initial_state()
    else:
        return []
@override(Policy)
def get_initial_state(self):
    if self.model:
        return self.model.get_initial_state()
    else:
        return []