Merge pull request #10 from cbg-ethz/feature/aggregate

feature/aggregate_locations
cbg-ethz · Nov 7, 2024 · 53a8b7f · 53a8b7f
2 parents fb46e91 + bfe1c4b
commit 53a8b7f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mutation_freq.py b/mutation_freq.py
@@ -131,6 +131,7 @@ def filter_by_location(data, location):
 
     # Dropdown to select a location
     locations = [
+        'Aggregate (All Locations)',
         'Lugano (TI)',
         'Zürich (ZH)',
         'Chur (GR)',
@@ -163,7 +164,8 @@ def filter_by_location(data, location):
         try:
             mutation_data = yaml.safe_load(mutation_config)
             filtered_data = filter_for_variant(tallymut, mutation_data)
-            filtered_data = filter_by_location(filtered_data, selected_location)
+            if selected_location != 'Aggregate (All Locations)':
+                filtered_data = filter_by_location(filtered_data, selected_location)
             plot_heatmap(filtered_data, location=selected_location)
         except yaml.YAMLError as e:
             st.error(f"Error parsing YAML: {e}")