moving dtypes and adding description

audeering · May 15, 2024 · 0d47e90 · 0d47e90
1 parent 5a8d9e8
commit 0d47e90
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/audinterface/core/segment.py b/audinterface/core/segment.py
@@ -515,6 +515,9 @@ def process_table(
     ) -> typing.Union[pd.Series, pd.DataFrame]:
         r"""Segment files or segments from a table.
 
+        The labels of the table
+        are reassigned to the new segments.
+
         If ``cache_root`` is not ``None``,
         a hash value is created from the index
         using :func:`audformat.utils.hash` and
@@ -570,15 +573,13 @@ def process_table(
         ends = []
         labels = []
         if isinstance(table, pd.Series):
-            dtype = table.dtype
             for j, ((file, start, _), index) in enumerate(y.items()):
                 files.extend([file] * len(index))
                 starts.extend(index.get_level_values("start") + start)
                 ends.extend(index.get_level_values("end") + start)
                 labels.extend([[table.iloc[j]] * len(index)])
             labels = np.hstack(labels)
         else:
-            dtypes = [table[col].dtype for col in table.columns]
             for j, ((file, start, _), index) in enumerate(y.items()):
                 files.extend([file] * len(index))
                 starts.extend(index.get_level_values("start") + start)
@@ -593,8 +594,10 @@ def process_table(
         index = audformat.segmented_index(files, starts, ends)
 
         if isinstance(table, pd.Series):
+            dtype = table.dtype
             table = pd.Series(labels, index, name=table.name, dtype=dtype)
         else:
+            dtypes = [table[col].dtype for col in table.columns]
             labels = {
                 col: pd.Series(
                     labels[:, icol], index=index, dtype=dtypes[icol]