From b21505bd8ca0dd0adab74a84698ed72e48a2f1f7 Mon Sep 17 00:00:00 2001
From: Glen Beane <356266+gbeane@users.noreply.github.com>
Date: Tue, 16 Jun 2026 13:53:00 -0400
Subject: [PATCH] allow pose versions greater than 6; change .apply(np.sum) to
 .sum() to work with newer versions of pandas

---
 src/jabs_postprocess/compare_gt.py     | 2 +-
 src/jabs_postprocess/utils/metadata.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/jabs_postprocess/compare_gt.py b/src/jabs_postprocess/compare_gt.py
index 3cdc47e..2456eb1 100644
--- a/src/jabs_postprocess/compare_gt.py
+++ b/src/jabs_postprocess/compare_gt.py
@@ -454,7 +454,7 @@ def generate_iou_scan(
     # Aggregate over animals
     performance_df = (
         performance_df.groupby(["stitch", "filter", "threshold"])[["tp", "fn", "fp"]]
-        .apply(np.sum)
+        .sum()
         .reset_index()
     )
     # Re-calculate PR/RE/F1
diff --git a/src/jabs_postprocess/utils/metadata.py b/src/jabs_postprocess/utils/metadata.py
index b18fbef..edf8871 100644
--- a/src/jabs_postprocess/utils/metadata.py
+++ b/src/jabs_postprocess/utils/metadata.py
@@ -5,7 +5,7 @@
 
 import yaml
 
-POSE_REGEX_STR = "_pose_est_v([2-6]).h5"
+POSE_REGEX_STR = "_pose_est_v([2-8]).h5"
 PREDICTION_REGEX_STR = "_behavior.h5"
 FEATURE_REGEX_STR = "features.h5"
 DATE_REGEX_STR = "[0-9]{4}-[0-9]{2}-[0-9]{2}"