Commit 85f8fc54 authored by hwangjeff's avatar hwangjeff Committed by Facebook GitHub Bot
Browse files

Add guards to prevent ffmpeg failures during dispatcher import (#3073)

Summary:
With the introduction of the backend dispatcher, importing torchaudio fails when ffmpeg is not available. This PR adds guards to resolve these failures.

Pull Request resolved: https://github.com/pytorch/audio/pull/3073

Reviewed By: NivekT, mthrok

Differential Revision: D43372870

Pulled By: hwangjeff

fbshipit-source-id: 7f6c2795430d7aeb742c2feb97984d5273f20aac
parent 2c9b3e59
...@@ -8,7 +8,9 @@ import torch ...@@ -8,7 +8,9 @@ import torch
import torchaudio.backend.soundfile_backend as soundfile_backend import torchaudio.backend.soundfile_backend as soundfile_backend
from torchaudio._extension import _FFMPEG_INITIALIZED, _SOX_INITIALIZED from torchaudio._extension import _FFMPEG_INITIALIZED, _SOX_INITIALIZED
from torchaudio.backend.common import AudioMetaData from torchaudio.backend.common import AudioMetaData
from torchaudio.io._compat import info_audio, info_audio_fileobj, load_audio, load_audio_fileobj, save_audio
if _FFMPEG_INITIALIZED:
from torchaudio.io._compat import info_audio, info_audio_fileobj, load_audio, load_audio_fileobj, save_audio
class Backend(ABC): class Backend(ABC):
......
# flake8: noqa # flake8: noqa
import torchaudio import torchaudio
from torchaudio._backend.utils import get_info_func, get_load_func, get_save_func
from . import utils from . import utils
from .utils import _is_backend_dispatcher_enabled, get_audio_backend, list_audio_backends, set_audio_backend from .utils import _is_backend_dispatcher_enabled, get_audio_backend, list_audio_backends, set_audio_backend
if _is_backend_dispatcher_enabled(): if _is_backend_dispatcher_enabled():
from torchaudio._backend.utils import get_info_func, get_load_func, get_save_func
torchaudio.info = get_info_func() torchaudio.info = get_info_func()
torchaudio.load = get_load_func() torchaudio.load = get_load_func()
torchaudio.save = get_save_func() torchaudio.save = get_save_func()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment