also use 'rate' attr from f['samples']

janclemenslab · Sep 22, 2023 · 2628adc · 2628adc
1 parent c7b71c0
commit 2628adc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/xarray_behave/__init__.py b/src/xarray_behave/__init__.py
@@ -1,5 +1,5 @@
 """xarray tools for behavioral data."""
-__version__ = "0.33.27"
+__version__ = "0.33.28"
 
 from .xarray_behave import assemble, assemble_metrics, load, save
 import os

diff --git a/src/xarray_behave/io/audio.py b/src/xarray_behave/io/audio.py
@@ -85,8 +85,11 @@ def load(
             if return_nonsong_channels:
                 non_song_channels = list(set(list(range(nb_channels))) - set(song_channels))
                 non_song = da[:, non_song_channels]
+
             if "rate" in f.attrs:
                 samplerate = f.attrs["rate"]
+            elif "rate" in f["samples"].attrs:
+                samplerate = f["samples"].attrs["rate"]
             else:
                 logger.info("   No sampling rate information in daq.h5 file - setting samplerate to default 10_000Hz.")
                 samplerate = 10_000
@@ -101,6 +104,8 @@ def load(
 
                 if "rate" in f.attrs:
                     samplerate = f.attrs["rate"]
+                elif "rate" in f["samples"].attrs:
+                    samplerate = f["samples"].attrs["rate"]
                 else:
                     logger.info("   No sampling rate information in daq.h5 file - setting samplerate to default 10_000Hz.")
                     samplerate = 10_000