Fix listify bug

huggingface · Sep 29, 2024 · f122941 · f122941
1 parent c7f6f51
commit f122941
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/datatrove/pipeline/filters/language_filter.py b/src/datatrove/pipeline/filters/language_filter.py
@@ -33,7 +33,7 @@ def __init__(
         super().__init__(exclusion_writer)
         self.language_threshold = language_threshold
         if isinstance(languages, str):
-            languages = list(languages)
+            languages = [languages]
         self.languages = languages
         self.backend = backend
         self.model = FT176LID(languages) if backend == "ft176" else GlotLID(languages)