From aaf174544644692cd294885299fa8df713178388 Mon Sep 17 00:00:00 2001 From: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> Date: Mon, 15 Jul 2024 18:30:02 +0200 Subject: [PATCH] Update services/worker/src/worker/job_runners/dataset/modalities.py Co-authored-by: Sylvain Lesage --- services/worker/src/worker/job_runners/dataset/modalities.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/services/worker/src/worker/job_runners/dataset/modalities.py b/services/worker/src/worker/job_runners/dataset/modalities.py index 4ebcaa599..a287a9c15 100644 --- 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'.")