Add missing comment (#2802)

huggingface · May 14, 2024 · 3760d8b · 3760d8b
1 parent af80a24
commit 3760d8b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/services/worker/src/worker/job_runners/dataset/compatible_libraries.py b/services/worker/src/worker/job_runners/dataset/compatible_libraries.py
@@ -504,7 +504,9 @@ def get_compatible_libraries_for_csv(dataset: str, hf_token: Optional[str], logi
         for loading_code in loading_codes:
             if len(loading_code["arguments"]["splits"]) == 1:
                 pattern = next(iter(loading_code["arguments"]["splits"].values()))
-                loading_code["code"] = DASK_CODE.format(function=function, dataset=dataset, pattern=pattern)
+                loading_code["code"] = DASK_CODE.format(
+                    function=function, dataset=dataset, pattern=pattern, comment=comment
+                )
             else:
                 loading_code["code"] = DASK_CODE_SPLITS.format(
                     function=function,