From 3760d8b7b3e84aed5f536ae8bda2ae1a1548f022 Mon Sep 17 00:00:00 2001 From: Andrea Francis Soria Jimenez Date: Tue, 14 May 2024 14:42:51 -0400 Subject: [PATCH] Add missing comment (#2802) --- .../src/worker/job_runners/dataset/compatible_libraries.py | 4 +++- 1 file changed, 3 insertions(+), 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 index 8632e3bd13..893cfae82a 100644 --- 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,