Fixed bug in parallel_text_corpus

sillsdev · Oct 22, 2024 · d8aa049 · d8aa049
1 parent b45a69b
commit d8aa049
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/machine/corpora/parallel_text_corpus.py b/machine/corpora/parallel_text_corpus.py
@@ -588,8 +588,8 @@ def count(self, include_empty: bool = True, text_ids: Optional[Iterable[str]] =
             if include_empty:
                 return len(self._df)
             return len(self._df[(self._df[self._source_column] != "") & (self._df[self._target_column] != "")])
-        return len(self._df[self._df[self._source_column].isin(text_ids)]) & (
-            len(self._df[self._target_column].isin(text_ids))
+        return len(self._df[self._df[self._source_column].isin(cast(Sequence[str], text_ids))]) & (
+            len(self._df[self._target_column].isin(cast(Sequence[str], text_ids)))
         )
 
     def _get_rows(self, text_ids: Optional[Iterable[str]] = None) -> Generator[ParallelTextRow, None, None]: