LUMIA-Group / rasat

The official implementation of the paper "RASAT: Integrating Relational Structures into Pretrained Seq2Seq Model for Text-to-SQL"(EMNLP 2022)
https://arxiv.org/abs/2205.06983
Apache License 2.0
63 stars 18 forks source link

problem with train/eval #22

Open SepehrAminiAfshar opened 1 year ago

SepehrAminiAfshar commented 1 year ago

Hi. I'm trying to run the project and produce the results myself. But I keep getting this error on train/eval commands.

Traceback (most recent call last): File "/content/rasat/seq2seq/run_seq2seq.py", line 292, in main() File "/content/rasat/seq2seq/run_seq2seq.py", line 158, in main metric, dataset_splits = load_dataset( File "/content/rasat/seq2seq/utils/dataset_loader.py", line 181, in load_dataset sparc_dataset_splits = prepare_splits( File "/content/rasat/seq2seq/utils/dataset.py", line 346, in prepare_splits train_split = _prepare_train_split( File "/content/rasat/seq2seq/utils/dataset.py", line 261, in _prepare_train_split relation_matrix_l = preprocess_by_dataset( File "/content/rasat/seq2seq/preprocess/choose_dataset.py", line 14, in preprocess_bydataset , relations = preprocessing_lgerels2t5rels_changeOrder(data_base_dir, dataset_name, t5_processed, mode, edge_type, use_coref, use_dependency) File "/content/rasat/seq2seq/preprocess/lgerels2t5rels_changeOrder.py", line 476, in preprocessing_lgerels2t5rels_changeOrder match_table_and_column(dataset_lgesql, table_lgesql, t5_tokenizer) File "/content/rasat/seq2seq/preprocess/lgerels2t5rels_changeOrder.py", line 136, in match_table_and_column raise e File "/content/rasat/seq2seq/preprocess/lgerels2t5rels_changeOrder.py", line 133, in match_table_and_column lge_table = table_lgesql[db_name]['table_names'] KeyError: 'sqlite_sequence:name,seq'

Does anybody have a solution?
If you have a working notebook that works just fine on train/eval commands, It would be kind of you to share it.

Thank you