Merge branch 'main' into kh-dev_snr-metrics

Author: hagikent

src/aind_dynamic_foraging_basic_analysis/__init__.py

@@ -1,6 +1,6 @@
 """Init package"""
 
-__version__ = "0.3.31"
+__version__ = "0.3.32"
 
 from .metrics.foraging_efficiency import compute_foraging_efficiency  # noqa: F401
 from .plot.plot_foraging_session import plot_foraging_session  # noqa: F401

src/aind_dynamic_foraging_basic_analysis/plot/plot_fip.py

@@ -425,7 +425,7 @@ def fip_psth_stats_plot(ax, stats_df, threshold=0.05):
     fip_psth_stats_plot(ax, etrs['stats'], threshold=0.05)
 
     """
-    unique_tests = stats_df["name"].unique()
+    unique_tests = np.unique(stats_df["name"].values)
     colors = style.get_colors(list(unique_tests), offset=0.25, cmap_name="plasma")
     for test in unique_tests:
         significant = stats_df.query("name == @test").query("p < @threshold")
@@ -613,7 +613,7 @@ def compute_hierarchical_error(
     # Organize results
     for index, val in enumerate(result.index.values):
         bootstraps[index]["time"] = val
-    result["hb_sem"] = [x["data_sem"] for x in bootstraps]
+    result["hb_sem"] = [x["{}_sem".format(data_column)] for x in bootstraps]
 
     return result, bootstraps