"tests/modeling/test_box2box_transform.py" did not exist on "5b3792fc3ef9ab6a6f8f30634ab2e52fb0941af3"
backend_utils.py 475 Bytes
Newer Older
moto's avatar
moto committed
1
2
3
4
5
6
7
import unittest

import torchaudio


def set_audio_backend(backend):
    """Allow additional backend value, 'default'"""
8
    backends = torchaudio.list_audio_backends()
moto's avatar
moto committed
9
    if backend == 'default':
10
        if 'sox_io' in backends:
11
            be = 'sox_io'
12
        elif 'soundfile' in backends:
moto's avatar
moto committed
13
14
15
16
17
18
19
            be = 'soundfile'
        else:
            raise unittest.SkipTest('No default backend available')
    else:
        be = backend

    torchaudio.set_audio_backend(be)