Updated parallel_text_corpus.py

sillsdev · Oct 24, 2024 · ba5ef17 · ba5ef17
1 parent df7dabe
commit ba5ef17
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
@@ -617,8 +617,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(set(text_ids))]) & (
-            len(self._df[self._target_column].isin(set(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]: