From e486b2d9ec86f592dec2b8253a7272d9bfc9f33a Mon Sep 17 00:00:00 2001 From: Raphael Townshend Date: Wed, 3 Feb 2021 15:08:41 -0800 Subject: [PATCH] Fix atom3d.datasets --- atom3d/datasets/__main__.py | 3 +-- atom3d/datasets/datasets.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/atom3d/datasets/__main__.py b/atom3d/datasets/__main__.py index 8c7fdd7..ad0688e 100644 --- a/atom3d/datasets/__main__.py +++ b/atom3d/datasets/__main__.py @@ -30,8 +30,7 @@ def main(input_dir, output_lmdb, filetype, score_path, serialization_format): fileext = 'xyz' else: fileext = filetype - file_list = fi.find_files(input_dir, fo.patterns[fileext]) - file_list.sort() + file_list = da.get_file_list(input_dir, fileext) logger.info(f'Found {len(file_list)} files.') dataset = da.load_dataset(file_list, filetype) diff --git a/atom3d/datasets/datasets.py b/atom3d/datasets/datasets.py index 3ef7c31..4f2f580 100644 --- a/atom3d/datasets/datasets.py +++ b/atom3d/datasets/datasets.py @@ -363,7 +363,7 @@ def get_file_list(input_path, filetype): file_list.append(x) else: file_list = fi.find_files(input_path, fo.patterns[filetype]) - return file_list + return sorted(file_list) def load_dataset(file_list, filetype, transform=None, include_bonds=False):