fixing linting issues

deepset-ai · Apr 29, 2024 · 89ffc21 · 89ffc21
1 parent a219d84
commit 89ffc21
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/haystack/document_stores/in_memory/document_store.py b/haystack/document_stores/in_memory/document_store.py
@@ -151,8 +151,8 @@ def _compute_tf(token: str, freq: Dict[str, int], doc_len: int) -> float:
             doc_len = doc_stats.doc_len
 
             score = 0.0
-            for tok in idf.keys():
-                score += idf[tok] * _compute_tf(tok, freq, doc_len)
+            for tok, idf_score in idf.keys():
+                score += idf_score * _compute_tf(tok, freq, doc_len)
             ret.append((doc, score))
 
         return ret
@@ -259,8 +259,8 @@ def _compute_tf(token: str, freq: Dict[str, int], doc_len: float) -> float:
             doc_len = doc_stats.doc_len
 
             score = 0.0
-            for tok in idf.keys():
-                score += idf[tok] * _compute_tf(tok, freq, doc_len)
+            for tok, idf_score in idf.keys():
+                score += idf_score * _compute_tf(tok, freq, doc_len)
             ret.append((doc, score))
 
         return ret