code: def set_states(self, states=None, value=None):
          
          if states is not None:
              assert value is None, "Only one of states & value can be specified."
              _load_general(states, self.state_arrays, (0,)*len(states))
          else:
              assert value is not None, "At least one of states & value must be specified."
              assert states is None, "Only one of states & value can be specified."
              for d_dst in self.state_arrays:
                  for dst in d_dst:
                      dst[:] = value
docs: Set value for states. Only one of states & value can be specified.
  
          Parameters
          ----------
          states : list of list of NDArrays
              source states arrays formatted like [[state1_dev1, state1_dev2],
              [state2_dev1, state2_dev2]].
          value : number
              a single scalar value for all state arrays.