fix clustering bug

dc0c1c31 · Koen van der Veen · 05b116c0 · dc0c1c31
Commit dc0c1c31 authored 2 years ago by Koen van der Veen
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/twitter_topic_model/plugin.py
+++ b/twitter_topic_model/plugin.py
@@ -83,7 +83,7 @@ class TwitterTopicModelPlugin(PluginBase):
        tweets_by_id = {tweet.id: tweet for tweet in tweets}

        for cluster in clusters:
-            cluster.tweets = [tweets_by_id[t.id] for t in tweets]
+            cluster.tweets = [tweets_by_id[t.id] for t in cluster.tweets]

        unassigned_tweets = [tweets_by_id[t.id] for t in unassigned_tweets]
        return clusters, unassigned_tweets