Specifying axis in plot_variants

blab · Jun 1, 2024 · 7519547 · 7519547
1 parent c32518a
commit 7519547
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evofr/plotting/plotting_primitives.py b/evofr/plotting/plotting_primitives.py
@@ -81,7 +81,7 @@ def plot_variants(
         for idx, variant in enumerate(variants_to_plot):
             variant_values = values[:, variant_indices[idx]]
 
-            median = np.median(variant_values)
+            median = np.median(variant_values, axis=0)
             for quantile in quantiles:
                 lower = np.quantile(variant_values, 0.5 * (1 - quantile), axis=0)
                 upper = np.quantile(variant_values, 0.5 * (1 + quantile), axis=0)