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

Update duckdb-nsql/eval/prompt_formatters.py

Browse files
duckdb-nsql/eval/prompt_formatters.py CHANGED
@@ -4,6 +4,7 @@ from random import shuffle
4
  from manifest import Manifest
5
  from schema import Table
6
  import re
 
7
 
8
 
9
  class RajkumarFormatter:
@@ -60,6 +61,7 @@ class RajkumarFormatter:
60
  context_text: str,
61
  ) -> str | list[str]:
62
  """Get prompt format."""
 
63
  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
64
 
65
  @classmethod
 
4
  from manifest import Manifest
5
  from schema import Table
6
  import re
7
+ import time
8
 
9
 
10
  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