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")}
def sanitize(result: JsonDict) -> JsonDict:
return {key: value for key, value in result.items() if key.startswith("best_span")}
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