Update services/worker/src/worker/job_runners/dataset/modalities.py

Co-authored-by: Sylvain Lesage <[email protected]>
huggingface · Jul 15, 2024 · aaf1745 · aaf1745
1 parent abaf965
commit aaf1745
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/services/worker/src/worker/job_runners/dataset/modalities.py b/services/worker/src/worker/job_runners/dataset/modalities.py
@@ -127,8 +127,8 @@ def detect_modalities_from_url_columns(dataset: str) -> set[DatasetModality]:
     Returns:
         `set[DatasetModality]`: A set of modalities.
     """
-    config_names_response = get_previous_step_or_raise(kind="dataset-split-names", dataset=dataset)
-    content = config_names_response["content"]
+    split_names_response = get_previous_step_or_raise(kind="dataset-split-names", dataset=dataset)
+    content = split_names_response["content"]
     if "splits" not in content and not isinstance(content["splits"], list):
         raise PreviousStepFormatError("Previous step did not return the expected content: 'splits'.")