in mujoco_py/mjsimstate.pyx [0:0]
def __eq__(self, other):
if not isinstance(other, self.__class__):
return NotImplemented
if set(self.udd_state.keys()) != set(other.udd_state.keys()):
return False
for k in self.udd_state.keys():
if isinstance(self.udd_state[k], Number) and self.udd_state[k] != other.udd_state[k]:
return False
elif not np.array_equal(self.udd_state[k], other.udd_state[k]):
return False
return (self.time == other.time and
np.array_equal(self.qpos, other.qpos) and
np.array_equal(self.qvel, other.qvel) and
np.array_equal(self.act, other.act))