Fix bug in extracting extension from filename -- was never loading h5…

… files. Closes #9
benlansdell · Feb 1, 2024 · 9327a4c · 9327a4c
1 parent 2b625c2
commit 9327a4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ethome/io.py b/ethome/io.py
@@ -77,7 +77,7 @@ def read_DLC_tracks(
     """
 
     ext = os.path.splitext(fn_in)[1]
-    if ext == "h5" or ext == "hdf5" or ext == "hdf":
+    if ext == ".h5" or ext == ".hdf5" or ext == ".hdf":
         df = pd.read_hdf(fn_in)
         ext = "h5"
     else: