From b9e13fe51fc421901f6430b1c0767ffa07b06247 Mon Sep 17 00:00:00 2001 From: Zhen Liu Date: Wed, 2 Aug 2023 02:17:28 +0200 Subject: [PATCH] Update save_meta.py --- metadata/save_meta.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metadata/save_meta.py b/metadata/save_meta.py index 7f6ed5d..0da62c5 100644 --- a/metadata/save_meta.py +++ b/metadata/save_meta.py @@ -9,7 +9,7 @@ if __name__ == "__main__": parser.add_argument('--extension', type=str, default='pt') args = parser.parse_args() - fpath_list = sorted([os.path.join(args.data_path, fname[:-len(args.extension)-1]) for fname in os.listdir(args.data_path) if fname.endswith('.' + args.extension)]) + fpath_list = sorted([os.path.join(args.data_path, fname) for fname in os.listdir(args.data_path) if fname.endswith('.' + args.extension)]) os.makedirs(args.json_path, exist_ok=True) json.dump(fpath_list, open(args.json_path, 'w'))