Merge pull request #153 from wintermelon008/dev

Fix bugs for quesnet_vec

Author: KenelmQLH

Diff for: 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 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'