mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-19 12:57:58 +00:00
Missing initialization
This commit is contained in:
parent
c927794248
commit
7d8123a526
1 changed files with 1 additions and 0 deletions
|
|
@ -343,6 +343,7 @@ class SQLQueryEnv(BaseEnv):
|
|||
scores["tokens"] = list()
|
||||
scores["masks"] = list()
|
||||
scores["scores"] = list()
|
||||
scores["inference_logprobs"] = list()
|
||||
|
||||
# Get table info from first item
|
||||
gold_sql = rollout_group_data[0]["gold_sql"]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue