silly error

allenai · Oct 24, 2024 · a4de365 · a4de365
1 parent 4f0ab10
commit a4de365
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/classifiers/src/dolma_classifiers/inference.py b/classifiers/src/dolma_classifiers/inference.py
@@ -152,7 +152,7 @@ def writer_worker(
 
             if element.source not in writers:
                 destination_path = source_destination_mapping[element.source]
-                writers[destination_path] = stack.enter_context(
+                writers[element.source] = stack.enter_context(
                     smart_open.open(destination_path, "wt", encoding="utf-8")
                 )
                 console_logger.info(f"Opened {destination_path} for writing")