fix daily/listening activity

metabrainz · Jan 28, 2025 · 5156d0b · 5156d0b
1 parent 01b3dd9
commit 5156d0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/listenbrainz_spark/stats/user/daily_activity.py b/listenbrainz_spark/stats/user/daily_activity.py
@@ -16,4 +16,4 @@ def get_daily_activity(stats_range: str, database: str = None) -> Iterator[Optio
     entity_obj = DailyActivityUserStatsQueryEntity(selector)
     message_creator = DailyActivityUserMessageCreator("user_daily_activity", selector, database)
     engine = IncrementalStatsEngine(entity_obj, message_creator)
-    return aggregator.run()
+    return engine.run()
diff --git a/listenbrainz_spark/stats/user/listening_activity.py b/listenbrainz_spark/stats/user/listening_activity.py
@@ -24,4 +24,4 @@ def get_listening_activity(stats_range: str, database: str = None)\
     entity_obj = ListeningActivityUserStatsQueryEntity(selector)
     message_creator = ListeningActivityUserMessageCreator("user_listening_activity", selector, database)
     engine = IncrementalStatsEngine(entity_obj, message_creator)
-    return aggregator.run()
+    return engine.run()