tdoehmen commited on
Commit
df3feb6
·
verified ·
1 Parent(s): 5a40a87

Update duckdb-nsql/eval/prompt_formatters.py

Browse files
duckdb-nsql/eval/prompt_formatters.py CHANGED
@@ -61,12 +61,12 @@ class RajkumarFormatter:
61
  context_text: str,
62
  ) -> str | list[str]:
63
  """Get prompt format."""
64
- time.sleep(10)
65
  return f"""{table_text}\n\n\n-- Using valid DuckDB SQL, answer the following question for the tables provided above.{context_text}\n\n-- {instruction}\n""" # noqa: E501
66
 
67
  @classmethod
68
  def format_model_output(cls, output_sql: str, prompt: str) -> str:
69
  """Format model output."""
 
70
  clean_sql = (output_sql
71
  .replace('sql\n', '')
72
  .replace('```sql\n', '')
 
61
  context_text: str,
62
  ) -> str | list[str]:
63
  """Get prompt format."""
 
64
  return f"""{table_text}\n\n\n-- Using valid DuckDB SQL, answer the following question for the tables provided above.{context_text}\n\n-- {instruction}\n""" # noqa: E501
65
 
66
  @classmethod
67
  def format_model_output(cls, output_sql: str, prompt: str) -> str:
68
  """Format model output."""
69
+ time.sleep(10)
70
  clean_sql = (output_sql
71
  .replace('sql\n', '')
72
  .replace('```sql\n', '')