Add files via upload

rfma23 · Aug 6, 2020 · 6136c4b · 6136c4b
1 parent 0e8accf
commit 6136c4b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/aertb/core/__init__.py b/aertb/core/__init__.py
@@ -1,4 +1,4 @@
 from .file_loader import FileLoader
 from .viz import make_gif
 from .types import event_dtype
-from .hdf5tools import HDF5FileIterator
+from .hdf5tools import HDF5FileIterator, HDF5File
diff --git a/aertb/core/hdf5tools.py b/aertb/core/hdf5tools.py
@@ -271,4 +271,4 @@ def fixed_train_test_split(self, n_train, n_test, rand=23):
             random.Random(rand).shuffle(train_samples)
             random.Random(rand + 1).shuffle(test_samples)
 
-        return HDF5FileIterator(self.file, train_samples), HDF5FileIterator(self.file, test_samples)
+        return HDF5FileIterator(self.file, train_samples), HDF5FileIterator(self.file, test_samples)
diff --git a/aertb/core/types.py b/aertb/core/types.py
@@ -21,6 +21,6 @@
                                ('x_', np.float32), ('y_', np.float32),
                                ('z_', np.float32), ('d_', np.float32)])
 
-Sample = namedtuple('Sample', ['name', 'label', 'events'])
+Sample = namedtuple('Sample', ['group', 'name'])
 EvSample = namedtuple('EvSample', ['label', 'name', 'events'])
 # =============================================================================