From 423510e98c4cc2b3d7ed33d4715e6246cf3b1fdc Mon Sep 17 00:00:00 2001 From: Ritu Gala Date: Thu, 18 Apr 2024 13:41:47 -0400 Subject: [PATCH] Update prompt2model/dataset_retriever/description_dataset_retriever.py Co-authored-by: Vijay Viswanathan --- prompt2model/dataset_retriever/description_dataset_retriever.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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