Fix atom3d.datasets

drorlab · Feb 3, 2021 · e486b2d · e486b2d
1 parent 3c4c357
commit e486b2d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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):