def state(self, state): if state is None: # or other checks? state = empty_array() if isinstance(state, sp.Expr): state = Array([state]) self.dim_state = len(state) self._state = state