change .h5 file loader to h5py

neuromorphs · Dec 11, 2023 · e139f40 · e139f40
1 parent 6bc0a0f
commit e139f40
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tonic/datasets/threeET_eyetracking.py b/tonic/datasets/threeET_eyetracking.py
@@ -1,8 +1,8 @@
 import os
 from typing import Any, Callable, Optional, Tuple
 
+import h5py
 import numpy as np
-import tables
 
 from tonic.dataset import Dataset
 from tonic.io import make_structured_array
@@ -78,8 +78,8 @@ def __getitem__(self, index: int) -> Tuple[Any, Any]:
  (events, target) where target is index of the target class.
  """
  # get events from .h5 file
- with tables.open_file(self.data[index], "r") as f:
- events = f.root.events.read()
+ with h5py.File(self.data[index], "r") as f:
+ events = f["events"][:]
  # load the sparse labels
  with open(self.targets[index], "r") as f:
  target = np.array(