fixed super() initialization error

DevKiHyun · Jun 20, 2017 · cdb321d · cdb321d
1 parent 60b89f0
commit cdb321d
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')