fix 'plot_stats_pyramid'

Alvaro · Alvaro · commit 85b84c4e297c · 2024-02-16T17:43:23.000-03:00
diff --git a/pyleida/_data_loader.py b/pyleida/_data_loader.py
@@ -1175,7 +1175,7 @@ def plot_stats_pyramid(self,metric='occupancies',conditions=None,despine=True):
                     ].reset_index(drop=True)
 
             dyn_metric = self._pool_dynamics_metric(metric=metric) 
-            dyn_metric = {k:v[v.condition.isin(conditions)] for k,v in dyn_metric.items()}
+            dyn_metric = {k:v[v.condition.isin(cond)] for k,v in dyn_metric.items()}
 
             stats_pyramid(
                 dyn_metric,
diff --git a/pyleida/_leida.py b/pyleida/_leida.py
@@ -1662,7 +1662,7 @@ def plot_stats_pyramid(self,metric='occupancies',conditions=None,despine=True):
                     ].reset_index(drop=True)
 
             dyn_metric = self._dynamics_[metric]
-            dyn_metric = {k:v[v.condition.isin(conditions)] for k,v in dyn_metric.items()}
+            dyn_metric = {k:v[v.condition.isin(cond)] for k,v in dyn_metric.items()}
 
             stats_pyramid(
                 dyn_metric,
