From 668fbe6d95b5efadfa588d8460c4e5946ca157fa Mon Sep 17 00:00:00 2001 From: tomguluson92 <314913739@qq.com> Date: Wed, 27 Jun 2018 09:43:03 +0800 Subject: [PATCH] bug fixed about self._partition --- lib/utils/peta_db.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/utils/peta_db.py b/lib/utils/peta_db.py index 98c03cb..abb58c2 100644 --- a/lib/utils/peta_db.py +++ b/lib/utils/peta_db.py @@ -55,7 +55,7 @@ def __init__(self, db_path, par_set_id): except NotImplementedError: import h5py print h5py.File(osp.join(self._db_path, 'partition.mat')).keys() - self.name = h5py.File(osp.join(self._db_path, 'partition.mat'))['partition'] + self._partition = h5py.File(osp.join(self._db_path, 'partition.mat'))['partition'] self.num_attr = self.name.shape[0] self.test_ind = None