diff --git a/prompt2model/dataset_retriever/description_dataset_retriever.py b/prompt2model/dataset_retriever/description_dataset_retriever.py index 5e5e852d9..91a425e0e 100644 --- a/prompt2model/dataset_retriever/description_dataset_retriever.py +++ b/prompt2model/dataset_retriever/description_dataset_retriever.py @@ -424,7 +424,7 @@ def make_dataset_from_samples( columns, "input_col" and "output_col". """ updated_inputs, updated_outputs = [], [] - if len(inputs) < 0 or len(inputs) != len(outputs): + if len(inputs) <= 0 or len(inputs) != len(outputs): logger.error("Length of inputs and outputs must be >0 and equal.") else: for i, o in zip(inputs, outputs):