Dimensionality inconsistant
I believe there is a typo in _explore_branch() in observations.py that leads to inconsistently sized output for observations. Line 441:
observation + ([-np.inf] * self.observation_dim) * num_cells_to_fill_in
should probably be an assignment to observation from what I understand.