Skip to content
This repository was archived by the owner on Dec 14, 2023. It is now read-only.

Commit bf74724

Browse files
committed
add remove_platform option for generate_and_store_maps
1 parent 22baefb commit bf74724

File tree

1 file changed

+10
-2
lines changed
  • apps/topics-map/src/python/topics_map

1 file changed

+10
-2
lines changed

apps/topics-map/src/python/topics_map/map.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -759,9 +759,17 @@ def store_map(db: DatabaseHandler,
759759
db.update_by_id('timespan_maps', timespan_map['timespan_maps_id'], {'url': url})
760760

761761

762-
def generate_and_store_maps(db: DatabaseHandler, timespans_id: int, memory_limit_mb: int) -> None:
762+
def generate_and_store_maps(
763+
db: DatabaseHandler,
764+
timespans_id: int,
765+
memory_limit_mb: int,
766+
remove_platforms: bool = True) -> None:
763767
"""Generate and layout graph and store various formats of the graph in timespans_maps."""
764-
graph = generate_and_layout_graph(db=db, timespans_id=timespans_id, memory_limit_mb=memory_limit_mb)
768+
graph = generate_and_layout_graph(
769+
db=db,
770+
timespans_id=timespans_id,
771+
memory_limit_mb=memory_limit_mb,
772+
remove_platforms=remove_platforms)
765773

766774
for color_by in ('community', 'retweet_partisanship', 'twitter_partisanship'):
767775
assign_colors(db=db, graph=graph, color_by=color_by)

0 commit comments

Comments
 (0)