Small speed up

euphorie · Jun 28, 2023 · 2e27982 · 2e27982
1 parent 5e95583
commit 2e27982
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/euphorie/content/browser/similar_titles.py b/src/euphorie/content/browser/similar_titles.py
@@ -157,16 +157,16 @@ def preprocess_text(line):
 
         # Group the lines that have a similarity score between
         # self.min_similarity and self.max_similarity
-        self.similar_brains = defaultdict(list)
+        similar_brains = self.similar_brains = defaultdict(list)
+        min_similarity = self.min_similarity
+        max_similarity = self.max_similarity
         for idx_x, brain_x in enumerate(brains):
             for idx_y, brain_y in enumerate(brains):
                 # skip pairs on the diagonal, they have obviously a similarity of 1
                 if idx_x != idx_y:
                     similarity = similarity_matrix[idx_x, idx_y]
-                    if self.min_similarity <= similarity <= self.max_similarity:
-                        self.similar_brains[brain_x].append(
-                            (brain_y, round(similarity, 2))
-                        )
+                    if min_similarity <= similarity <= max_similarity:
+                        similar_brains[brain_x].append((brain_y, round(similarity, 2)))
 
     @button.buttonAndHandler(_("Search"))
     def handle_search(self, action):