def construct_query2box(self, name):
    """Load query2box data into a list of examples."""
    task = name.split('_')[-1]
    # we don't train our model in the same way as query2box,
    # so we won't load their training data.
    test_data = pickle.load(
        gfile.GFile(self.root_dir + 'test_ans_%s.pkl' % task, 'rb'))

    converted_test_data = list()
    for query, unused_answers in test_data.items():
      converted_test_data.append(query)

    return None, converted_test_data
Load query2box data into a list of examples.
language/emql/data_loader.py