diff --git a/prompt2model/dataset_retriever/description_dataset_retriever.py b/prompt2model/dataset_retriever/description_dataset_retriever.py index c6e3047b6..15eb308ac 100644 --- a/prompt2model/dataset_retriever/description_dataset_retriever.py +++ b/prompt2model/dataset_retriever/description_dataset_retriever.py @@ -265,7 +265,7 @@ def automatic_column_selection( output_column = response["output"] if len(input_columns) < 1 or len(output_column) != 1: raise RuntimeError( - f"Incorrect number of cols: {input_columns}, {output_column} " + f"Incorrect number of columns: {input_columns}, {output_column} " ) # noqa: E501 dataset_columns = dataset_columns