def states_type()

in simuleval/agents/agent.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def states_type(self, args):
        if self.data_type == "text":
            return TextStates
        elif self.data_type == "speech":
            return SpeechStates
        else:
            raise NotImplementedError