chore: skip unnecessary key checks prior to accessing a dictionary (#4497)

This commit is contained in:
Bowen Liang 2024-05-19 18:30:45 +08:00 • committed by GitHub
commit 04ad46dd31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
30 changed files with 45 additions and 44 deletions

View file

@ -124,7 +124,7 @@ class DatasetRetrieval:
document_score_list = {}
for item in all_documents:
if 'score' in item.metadata and item.metadata['score']:
if item.metadata.get('score'):
document_score_list[item.metadata['doc_id']] = item.metadata['score']
document_context_list = []