lint

Roberto Alsina · Roberto Alsina · commit 0516994b2f42 · 2017-05-22T17:11:14.000-03:00
diff --git a/v7/similarity/similarity.py b/v7/similarity/similarity.py
@@ -30,33 +30,32 @@
 
 from nikola.plugin_categories import Task
 
+
 class Similarity(Task):
     """Calculate post similarity."""
     name = "similarity"
 
     def set_site(self, site):
         self.site = site
-        
+
     def gen_tasks(self):
         """Build similarity data for each post."""
         self.site.scan_posts()
-        
+
         texts = []
-        
+
         for p in self.site.timeline:
             texts.append(p.text(strip_html=True).lower().split())
-        
+
         dictionary = gensim.corpora.Dictionary(texts)
         corpus = [dictionary.doc2bow(text) for text in texts]
         lsi = gensim.models.LsiModel(corpus, id2word=dictionary, num_topics=2)
         index = gensim.similarities.MatrixSimilarity(lsi[corpus])
-        
+
         for i, post in enumerate(self.site.timeline):
             doc = texts[i]
             vec_bow = dictionary.doc2bow(doc)
             vec_lsi = lsi[vec_bow]
             sims = index[vec_lsi]
             sims = sorted(enumerate(sims), key=lambda item: -item[1])
             print(i, sims[:10])
-        
-                         
