From f7ac38b93adcf0272cfe1ecf59f0bd3e69e158ef Mon Sep 17 00:00:00 2001 From: saundersp Date: Sun, 28 Apr 2024 00:26:31 +0200 Subject: [PATCH] python : added comment for potential indices in unit test --- python/common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/common.py b/python/common.py index 7acfffe..966ff4b 100644 --- a/python/common.py +++ b/python/common.py @@ -60,11 +60,13 @@ def unit_test(TS: List[int], labels: List[str] = ['CPU', 'GPU', 'PY', 'PGPU'], t title = f'X_{set_name}_feat_argsort' print(f'Loading {title}...', file = stderr, end = '\r') feat = None + #indices = pickle_multi_loader(['indices'], OUT_DIR)[0] bs = [] for label in labels: if feat is None: feat_tmp = pickle_multi_loader([f'X_{set_name}_feat_{label}'], OUT_DIR)[0] if feat_tmp is not None: + #feat = feat_tmp[indices] feat = feat_tmp bs.append(pickle_multi_loader([f'{title}_{label}'], OUT_DIR)[0])