d02dff76c1e8fcfcff277a4ea0a65f9c8259605e,torchaudio/__init__.py,,save,#Any#Any#Any#,37

Before Change

    assert not src.is_cuda

    filename, extension = os.path.splitext(filepath)
    assert type(sample_rate) == int

    if isinstance(src, torch.FloatTensor):
        func = th_sox.libthsox_Float_write_audio_file
    elif isinstance(src, torch.DoubleTensor):

After Change

def save(filepath, src, sample_rate):
    filename, extension = os.path.splitext(filepath)
    if type(sample_rate) != int:
        raise TypeError("Sample rate should be a integer")

    check_input(src)
    typename = type(src).__name__.replace("Tensor", "")
    func = getattr(th_sox, "libthsox_{}_write_audio_file".format(typename))

    func(bytes(filepath), src, extension[1:], sample_rate)

Instances