conditional legend on growth rates

micom-dev · Sep 12, 2024 · 7c3f84a · 7c3f84a
1 parent d5af552
commit 7c3f84a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/micom/data/templates/growth.html b/micom/data/templates/growth.html
@@ -84,7 +84,7 @@ <h1 class="title">
               selection: "pts",
               field: "{{ color }}",
               type: "nominal",
-              legend: null
+              {% if color == "taxon" %}legend: null{% endif %}
             },
             value: "lightgrey"
           },

diff --git a/tests/test_viz.py b/tests/test_viz.py
@@ -10,9 +10,10 @@
 
 def random_groups(res):
     """Create some random groups for samples."""
+    sids = res.growth_rates.sample_id.unique()
     groups = pd.Series(
-        random.choices(["a", "b"], k=4),
-        index=res.growth_rates.sample_id.unique(),
+        random.choices(["a", "b"], k=len(sids)),
+        index=sids,
         name="random",
     )
     return groups