def _get_entity_literals(self, question_data: JsonDict) -> JsonDict: res: JsonDict = {} for key, value in question_data.items(): if '_literals' in key and key.replace('_literals', '') in self._entity_types: res.update(value) return res
def sanitize(result: JsonDict) -> JsonDict: return {key: value for key, value in result.items() if key.startswith("best_span")}