def validate_output_state_traits(self, out):
if self.multi_output:
if not isinstance(out, Sequence):
raise StateDimensionalityError(
"state sequence required on output")
for state in out:
for trait in self.outputs:
self.validate_state_trait(state, trait, "output")
else:
if not isinstance(out, Mapping):
raise StateDimensionalityError(
"single state required on output")
for trait in self.outputs:
self.validate_state_trait(out, trait, "output")
def validate_input_state_traits(self, inp):
if self.multi_input:
if not isinstance(inp, Sequence):
raise StateDimensionalityError(
"state sequence required on input")
for state in inp:
for trait in self.inputs:
self.validate_state_trait(state, trait, "input")
else:
if not isinstance(inp, Mapping):
raise StateDimensionalityError(
"single state required on input")
for trait in self.inputs:
self.validate_state_trait(inp, trait, "input")