From fc52efcc74509e2a1987046c39252d9ba9dc993e Mon Sep 17 00:00:00 2001 From: danleifeng Date: Thu, 6 Feb 2020 15:26:09 +0000 Subject: [PATCH] add format for test reader --- plsc/entry.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/plsc/entry.py b/plsc/entry.py index 12ad168..c5da675 100644 --- a/plsc/entry.py +++ b/plsc/entry.py @@ -844,7 +844,8 @@ class Entry(object): test_reader = self.test_reader if not self.test_initialized: test_list, test_name_list = test_reader(self.dataset_dir, - self.val_targets) + self.val_targets, + data_format=self.data_format) assert self._get_info('test_list') is None assert self._get_info('test_name_list') is None self._set_info('test_list', test_list) -- GitLab