Update prompt2model/dataset_retriever/description_dataset_retriever.py

Co-authored-by: Vijay Viswanathan <[email protected]>
neulab · Apr 18, 2024 · 547d4a3 · 547d4a3
1 parent 92f127a
commit 547d4a3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/prompt2model/dataset_retriever/description_dataset_retriever.py b/prompt2model/dataset_retriever/description_dataset_retriever.py
@@ -649,7 +649,8 @@ def get_datasets_of_required_size(
 
             if config_name is None:
                 del datasets_info[dataset_name]
-                continue  # If it couldn't find a relevant config, delete the entire dataset. # noqa: E501
+                # If it couldn't find a relevant config, delete the entire dataset. noqa: E501
+                continue
 
             dataset_info = datasets_info[dataset_name]["configs"][config_name]
             canonicalized_dataset = self.canonicalize_dataset_automatically(