Commit d418c4d2 authored by Yutong Zhao's avatar Yutong Zhao
Browse files

Make anitools python2 friendly

parent 561a239f
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
# Written by Roman Zubatyuk and Justin S. Smith
# Modified by Yutong Zhao to make python2 compatible
import h5py
import numpy as np
import platform
@@ -70,7 +71,8 @@ class anidataloader(object):

        yield data
      else:  # test for group (go down)
        yield from self.h5py_dataset_iterator(item, path)
        for s in self.h5py_dataset_iterator(item, path):
          yield s

  ''' Default class iterator (iterate through all data) '''