From 94e44738223c4316d58fac6a5bec0dec24586805 Mon Sep 17 00:00:00 2001
From: Pablo Mavrocordatos <pablo.mavrocordatos@hotmail.ch>
Date: Wed, 13 Aug 2025 17:05:50 +0200
Subject: [PATCH] Add support for DLC .h5 files in data reader

---
 src/lisbet/io/core.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lisbet/io/core.py b/src/lisbet/io/core.py
index e40f342..6c7303a 100644
--- a/src/lisbet/io/core.py
+++ b/src/lisbet/io/core.py
@@ -69,7 +69,7 @@ def _load_posetracks(seq_path, data_format, data_scale, select_coords, rename_co
     # Valid filenames and their corresponding loading functions
     # TODO: Test re matching for all supported formats.
     data_readers = {
-        "DLC": (r"(?i)(DLC.*?shuffle\d+|tracking).*\.csv$", load_poses.from_dlc_file),
+        "DLC": (r"(?i)(DLC.*?shuffle\d+|tracking).*\.(csv|h5)$", load_poses.from_dlc_file),
         "SLEAP": (r"(?i)SLEAP.*\.h5$", load_poses.from_sleap_file),
         "movement": (r"(?i)tracking.*\.nc$", partial(xr.open_dataset, engine="scipy")),
     }