Bugfix make_dataset

drorlab · Feb 2, 2021 · 7f2ab96 · 7f2ab96
1 parent 673d4d2
commit 7f2ab96
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/atom3d/datasets/__main__.py b/atom3d/datasets/__main__.py
@@ -36,8 +36,7 @@ def main(input_dir, output_lmdb, filetype, score_path, serialization_format):
 
     dataset = da.load_dataset(file_list, filetype)
     da.make_lmdb_dataset(
-        dataset, output_lmdb, filetype,
-        serialization_format=serialization_format)
+        dataset, output_lmdb, serialization_format=serialization_format)
 
 
 if __name__ == "__main__":