def dimension()

in src/diarizers/models/pyannet.py [0:0]

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

    def dimension(self) -> int:
        """Dimension of output"""
        if isinstance(self.specifications, tuple):
            raise ValueError("PyanNet does not support multi-tasking.")

        if self.specifications.powerset:
            return self.specifications.num_powerset_classes
        else:
            return len(self.specifications.classes)