def output_types(self): """Returns definitions of module output ports. """ return { "processed_signal": NeuralType(('B', 'D', 'T'), MFCCSpectrogramType()), "processed_length": NeuralType(tuple('B'), LengthsType()), }
def test_inheritance_spec_augment_example(self): input = NeuralType(('B', 'D', 'T'), SpectrogramType()) out1 = NeuralType(('B', 'D', 'T'), MelSpectrogramType()) out2 = NeuralType(('B', 'D', 'T'), MFCCSpectrogramType()) assert out1.compare(out2) == NeuralTypeComparisonResult.INCOMPATIBLE assert out2.compare(out1) == NeuralTypeComparisonResult.INCOMPATIBLE assert input.compare(out1) == NeuralTypeComparisonResult.GREATER assert input.compare(out2) == NeuralTypeComparisonResult.GREATER assert out1.compare(input) == NeuralTypeComparisonResult.LESS assert out2.compare(input) == NeuralTypeComparisonResult.LESS