pyannote.database 5.1.0__tar.gz → 5.1.2__tar.gz

{pyannote.database-5.1.0/pyannote.database.egg-info → pyannote.database-5.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyannote.database
-Version: 5.1.0
+Version: 5.1.2
 Summary: Interface to multimedia databases and experimental protocols
 Home-page: http://pyannote.github.io/
 Author: Hervé Bredin

{pyannote.database-5.1.0 → pyannote.database-5.1.2}/pyannote/database/_version.py

@@ -11,8 +11,8 @@ version_json = '''
 {
  "dirty": false,
  "error": null,
- "full-revisionid": "b1fc3492e7ea8aa822f1c0993fcd38c485a6b61e",
- "version": "5.1.0"
+ "full-revisionid": "d8345c4e4df103e425a9d4dfd512da4c1817640a",
+ "version": "5.1.2"
 }
 '''  # END VERSION_JSON
 

{pyannote.database-5.1.0 → pyannote.database-5.1.2}/pyannote/database/custom.py

@@ -49,17 +49,18 @@ import string
 from . import protocol as protocol_module
 
 from pyannote.database.protocol.protocol import ProtocolFile
-import yaml
+
+
 import warnings
 from numbers import Number
 from typing import Text, Dict, Callable, Any, Union
 import functools
 
-from .protocol.protocol import Subset, Scope
+from .protocol.protocol import Subset
 from .protocol.segmentation import SegmentationProtocol
 from .protocol.speaker_diarization import SpeakerDiarizationProtocol
 
-import pkg_resources
+from importlib.metadata import entry_points
 
 from .util import get_annotated
 
@@ -68,7 +69,7 @@ from .loader import load_lst, load_trial
 # All "Loader" classes types (eg RTTMLoader, UEMLoader, ...) retrieved from the entry point.
 LOADERS = {
     ep.name: ep
-    for ep in pkg_resources.iter_entry_points(group="pyannote.database.loader")
+    for ep in entry_points(group="pyannote.database.loader")
 }
 
 

{pyannote.database-5.1.0 → pyannote.database-5.1.2}/pyannote/database/loader.py

@@ -90,7 +90,7 @@ def load_trial(file_trial):
     """
 
     trials = pd.read_table(
-        file_trial, delim_whitespace=True, names=["reference", "uri1", "uri2"]
+        file_trial, sep="\s+", names=["reference", "uri1", "uri2"]
     )
 
     for _, reference, uri1, uri2 in trials.itertuples():
@@ -289,7 +289,7 @@ class CTMLoader:
             "confidence": float,
         }
         self.data_ = pd.read_csv(
-            ctm, names=names, dtype=dtype, delim_whitespace=True
+            ctm, names=names, dtype=dtype, sep="\s+"
         ).groupby("uri")
 
     def __call__(self, current_file: ProtocolFile) -> Union["Doc", None]:
@@ -354,7 +354,7 @@ class MAPLoader:
             "uri": str,
         }
         self.data_ = pd.read_csv(
-            mapping, names=names, dtype=dtype, delim_whitespace=True
+            mapping, names=names, dtype=dtype, sep="\s+"
         )
 
         # get colum 'value' dtype, allowing us to acces it during subset

{pyannote.database-5.1.0 → pyannote.database-5.1.2}/pyannote/database/util.py

@@ -78,17 +78,17 @@ def get_unique_identifier(item):
 def get_annotated(current_file):
     """Get part of the file that is annotated.
 
-        Parameters
-        ----------
-        current_file : `dict`
-            File generated by a `pyannote.database` protocol.
-
-        Returns
-        -------
-        annotated : `pyannote.core.Timeline`
-            Part of the file that is annotated. Defaults to
-            `current_file["annotated"]`. When it does not exist, try to use the
-            full audio extent. When that fails, use "annotation" extent.
+    Parameters
+    ----------
+    current_file : `dict`
+        File generated by a `pyannote.database` protocol.
+
+    Returns
+    -------
+    annotated : `pyannote.core.Timeline`
+        Part of the file that is annotated. Defaults to
+        `current_file["annotated"]`. When it does not exist, try to use the
+        full audio extent. When that fails, use "annotation" extent.
     """
 
     # if protocol provides 'annotated' key, use it
@@ -179,7 +179,7 @@ def load_rttm(file_rttm, keep_type="SPEAKER"):
         file_rttm,
         names=names,
         dtype=dtype,
-        delim_whitespace=True,
+        sep="\s+",
         keep_default_na=True,
     )
 
@@ -213,7 +213,7 @@ def load_stm(file_stm):
     dtype = {"uri": str, "speaker": str, "start": float, "end": float}
     data = pd.read_csv(
         file_stm,
-        delim_whitespace=True,
+        sep="\s+",
         usecols=[0, 2, 3, 4],
         dtype=dtype,
         names=list(dtype),
@@ -250,7 +250,7 @@ def load_mdtm(file_mdtm):
         file_mdtm,
         names=names,
         dtype=dtype,
-        delim_whitespace=True,
+        sep="\s+",
         keep_default_na=False,
     )
 
@@ -281,7 +281,7 @@ def load_uem(file_uem):
 
     names = ["uri", "NA1", "start", "end"]
     dtype = {"uri": str, "start": float, "end": float}
-    data = pd.read_csv(file_uem, names=names, dtype=dtype, delim_whitespace=True)
+    data = pd.read_csv(file_uem, names=names, dtype=dtype, sep="\s+")
 
     timelines = dict()
     for uri, parts in data.groupby("uri"):
@@ -306,7 +306,7 @@ def load_lab(path, uri: str = None) -> Annotation:
 
     names = ["start", "end", "label"]
     dtype = {"start": float, "end": float, "label": str}
-    data = pd.read_csv(path, names=names, dtype=dtype, delim_whitespace=True)
+    data = pd.read_csv(path, names=names, dtype=dtype, sep="\s+")
 
     annotation = Annotation(uri=uri)
     for i, turn in data.iterrows():
@@ -388,7 +388,6 @@ class LabelMapper(object):
         self.keep_missing = keep_missing
 
     def __call__(self, current_file):
-
         if not self.keep_missing:
             missing = set(current_file["annotation"].labels()) - set(self.mapping)
             if missing and not self.keep_missing:

{pyannote.database-5.1.0 → pyannote.database-5.1.2/pyannote.database.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyannote.database
-Version: 5.1.0
+Version: 5.1.2
 Summary: Interface to multimedia databases and experimental protocols
 Home-page: http://pyannote.github.io/
 Author: Hervé Bredin