Merge pull request #16 from sanchit-gandhi/v0.2.1-release

Release: v0.2.1

README.md

@@ -11,7 +11,7 @@
 
 # Installation
 
-**With `pip`** (official package)
+With `pip` (official package)
 
 ```bash
 pip install speechbox

setup.py

@@ -175,7 +175,7 @@ def run(self):
 
 setup(
     name="speechbox",
-    version="0.2.0",  # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
+    version="0.2.1",  # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
     description="Speechbox",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

src/speechbox/__init__.py

@@ -22,7 +22,7 @@
 # to defer the actual importing for when the objects are requested. This way `import transformers` provides the names
 # in the namespace without actually importing anything (and especially none of the backends).
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from .utils import (is_accelerate_available, is_pyannote_available,
                     is_scipy_available, is_torchaudio_available,

src/speechbox/utils/dummy_transformers_and_torchaudio_and_pyannote_objects.py

@@ -5,15 +5,15 @@
 
 
 class ASRDiarizationPipeline(metaclass=DummyObject):
-    _backends = ["transformers", "torchaudio", "pyannote"]
+    _backends = ["transformers", "torchaudio", "pyannote.audio"]
 
     def __init__(self, *args, **kwargs):
-        requires_backends(self, ["transformers", "torchaudio", "pyannote"])
+        requires_backends(self, ["transformers", "torchaudio", "pyannote.audio"])
 
     @classmethod
     def from_config(cls, *args, **kwargs):
-        requires_backends(cls, ["transformers", "torchaudio", "pyannote"])
+        requires_backends(cls, ["transformers", "torchaudio", "pyannote.audio"])
 
     @classmethod
     def from_pretrained(cls, *args, **kwargs):
-        requires_backends(cls, ["transformers", "torchaudio", "pyannote"])
+        requires_backends(cls, ["transformers", "torchaudio", "pyannote.audio"])

src/speechbox/utils/import_utils.py

@@ -61,7 +61,7 @@
 _torchaudio_available = importlib.util.find_spec("torchaudio") is not None
 try:
     _ = importlib_metadata.version("torchaudio")
-    _torchaudio_metadata = importlib_metadata.metadata("orchaudio")
+    _torchaudio_metadata = importlib_metadata.metadata("torchaudio")
 except importlib_metadata.PackageNotFoundError:
     _torchaudio_available = False
 

utils/check_dummies.py

@@ -116,6 +116,7 @@ def create_dummy_files(backend_specific_objects=None):
 
     for backend, objects in backend_specific_objects.items():
         backend_name = "[" + ", ".join(f'"{b}"' for b in backend.split("_and_")) + "]"
+        backend_name = backend_name.replace("pyannote", "pyannote.audio")
         dummy_file = "# This file is autogenerated by the command `make fix-copies`, do not edit.\n"
         dummy_file += "# flake8: noqa\n\n"
         dummy_file += "from ..utils import DummyObject, requires_backends\n\n"