Commit b49dbdfb authored by Marek Medved's avatar Marek Medved
Browse files

fix --simple output

parent 31a0bf31

query_database.py

+7 −7
Original line number Diff line number Diff line
@@ -91,17 +91,17 @@ def print_record(db, record_id): 


    print('question:')

    for i in get_content(record.question, vocabulary, part='w'):

        print(f'\ts: {" ".join(i)}')

        print(f'\ts: {" ".join([x["word"] for x in i])}')



    print('a_sel:')

    for i in get_content(record.answer_selection, vocabulary, part='w'):

        print(f'\ts: {" ".join(i)}')

        print(f'\ts: {" ".join([x["word"] for x in i])}')



    print(f'a_sel_pos: {record.text_answer_position}')



    print('a_ext:')

    for i in get_content(record.answer_extraction, vocabulary, part='w'):

        print(f'\ts: {" ".join(i)}')

        print(f'\ts: {" ".join([x["word"] for x in i])}')



    print('similar_answers:')

    for key, value in  record.similar_answers.items():
@@ -112,15 +112,15 @@ def print_record(db, record_id): 


    print(f'text_title:')

    for i in get_content(kb.url2doc.get(record.text)["title"], vocabulary, part="w"):

        print(f'\ts: {" ".join(i)}')

        print(f'\ts: {" ".join([x["word"] for x in i])}')



    print('text:')

    for sent_and_phrs in get_content_ctx(record.text, kb, vocabulary, part='w'):

        print(f'\ts: {" ".join(sent_and_phrs["sent"])}')

    for idx, sent_and_phrs in enumerate(get_content_ctx(record.text, kb, vocabulary, part='w')):

        print(f'\ts_{idx}: {" ".join([x["word"] for x in sent_and_phrs["sent"]])}')

        for key, phrs in sent_and_phrs['ctx'].items():

            print(f'\t\tctx_type: {key}')

            for p in phrs:

                print(f'\t\t\tc: {" ".join(p)}')

                print(f'\t\t\tc: {" ".join([x["word"] for x in p])}')





def main():