From ace19601f407a6013964b675d683fbb4a8a1c70b Mon Sep 17 00:00:00 2001 From: wintermelon008 <1904346407@qq.com> Date: Fri, 13 Oct 2023 10:24:52 +0800 Subject: [PATCH 1/2] Fix bugs for quesnet_vec --- EduNLP/Pretrain/quesnet_vec.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/EduNLP/Pretrain/quesnet_vec.py b/EduNLP/Pretrain/quesnet_vec.py index d6b0ae78..fecbce61 100644 --- a/EduNLP/Pretrain/quesnet_vec.py +++ b/EduNLP/Pretrain/quesnet_vec.py @@ -157,14 +157,14 @@ def _convert_to_ids(self, item: Union[str, dict, list], key=lambda x: x, token_item = self.tokenize(item, key) token_idx = [] for _, w in enumerate(token_item): - if isinstance(w, FigureSegment) and 'ques_figure_ids' in item.keys(): + if isinstance(w, FigureSegment) and isinstance(item, dict) and 'ques_figure_ids' in item.keys(): # image try: fig_id = f"{w.src[10:-1]}" fig_index = item['ques_figure_ids'].index(fig_id) - if self.img_dir != "": + if self.img_dir != None: fig_src = os.path.join(self.img_dir, fig_id) if '.png' in item['ques_figure_paths'][fig_index]: fig_src += '.png' From d65242c5603e623d9d8df9a55fa7cee9bc059e4e Mon Sep 17 00:00:00 2001 From: wintermelon008 <1904346407@qq.com> Date: Fri, 13 Oct 2023 11:12:02 +0800 Subject: [PATCH 2/2] Update --- EduNLP/Pretrain/quesnet_vec.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/EduNLP/Pretrain/quesnet_vec.py b/EduNLP/Pretrain/quesnet_vec.py index fecbce61..a331157b 100644 --- a/EduNLP/Pretrain/quesnet_vec.py +++ b/EduNLP/Pretrain/quesnet_vec.py @@ -164,7 +164,7 @@ def _convert_to_ids(self, item: Union[str, dict, list], key=lambda x: x, fig_id = f"{w.src[10:-1]}" fig_index = item['ques_figure_ids'].index(fig_id) - if self.img_dir != None: + if self.img_dir is not None: fig_src = os.path.join(self.img_dir, fig_id) if '.png' in item['ques_figure_paths'][fig_index]: fig_src += '.png'