tdoehmen commited on
Commit
4a4649a
·
verified ·
1 Parent(s): d3a4ed6

Update duckdb-nsql/eval/prompt_formatters.py

Browse files
duckdb-nsql/eval/prompt_formatters.py CHANGED
@@ -66,6 +66,7 @@ class RajkumarFormatter:
66
  def format_model_output(cls, output_sql: str, prompt: str) -> str:
67
  """Format model output."""
68
  clean_sql = (output_sql
 
69
  .replace('```sql\n', '')
70
  .replace('```duckdb\n', '')
71
  .replace('```\n', '')
 
66
  def format_model_output(cls, output_sql: str, prompt: str) -> str:
67
  """Format model output."""
68
  clean_sql = (output_sql
69
+ .replace('sql\n', '')
70
  .replace('```sql\n', '')
71
  .replace('```duckdb\n', '')
72
  .replace('```\n', '')