DAMO-NLP-SG / DAMO-NLP-SG/LLM-R2

IndexError: list index out of range

Open
#6 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
54
Forks
13
PR merge metrics
No merged PRs in 30d

Description

Hi,When I try to ```pyton LLM_R2.py``` to get the query rewrite result from JOB benchmark, the code meets an error:
```
#########################################
this code is running on: cuda:0
#########################################
Traceback (most recent call last):
File "/home/syy/LLM-R2/src/LLM_R2.py", line 720, in
LLM_R2(dataset, method, num_promos)
File "/home/syy/LLM-R2/src/LLM_R2.py", line 542, in LLM_R2
promo_pool_pos = get_pool('/home/syy/LLM-R2/data/data_llmr2/pools/pos_pool_' + dataset + '_updated.csv', method)
File "/home/syy/LLM-R2/src/LLM_R2.py", line 514, in get_pool
embeddings = batcher(batch2, pool_df['db_id'].tolist())
File "/home/syy/LLM-R2/src/LLM_R2.py", line 54, in batcher
sent_features = prepare_enc_data(sentences, pre_lang_model, db_ids)
File "/home/syy/LLM-R2/src/encoder.py", line 324, in prepare_enc_data
nodes = [traversePlan(get_physical_tree(db_id, sql_input)) for sql_input in query_dataset[i]]
File "/home/syy/LLM-R2/src/encoder.py", line 324, in
nodes = [traversePlan(get_physical_tree(db_id, sql_input)) for sql_input in query_dataset[i]]
File "/home/syy/LLM-R2/src/encoder.py", line 73, in traversePlan
root_dict = root[0]
IndexError: list index out of range
```
I'm sure the input path and ```dataset``` in LLM_R2.py were settled correctly, because I have sucessfully obtained TPC-H and DSB benchmark query rewrite results. I'm not sure why this error happens.

In addition, I find JOB benchmark queries in ```/LLM-R2/data/data_llmr2/pools/neg_pool_job_syn_updated.csv、/LLM-R2/data/data_llmr2/pools/pos_pool_job_syn_updated.csv、/LLM-R2/data/data_llmr2/queries/queries_job_syn_test.csv```... etc. were different from the offical JOB benchmark queries :https://www.vldb.org/pvldb/vol9/p204-leis.pdf.[1] Could you help me find your query source to get a better understand?

[1]Viktor Leis,etc. How Good Are Query Optimizers, Really?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.