diff --git a/prompt2model/dataset_retriever/description_dataset_retriever.py b/prompt2model/dataset_retriever/description_dataset_retriever.py
index 15eb308ac..e3a4ecfb0 100644
--- 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(