Merge pull request SeanNaren#98 from aaronzira/master

fixed super() initialization error
DevKiHyun · Jun 20, 2017 · 8ffbcf5 · 8ffbcf5
2 parents 60b89f0 + cdb321d
commit 8ffbcf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data/bucketing_sampler.py b/data/bucketing_sampler.py
@@ -28,7 +28,7 @@ def __init__(self, data_source):
  memory overhead.
  :param data_source: The dataset to be sampled from
  """
- super().__init__(data_source)
+ super(BucketingSampler, self).__init__(data_source)
  self.data_source = data_source
  assert hasattr(self.data_source, 'bins_to_samples')