From 7f2ab9648ecf2bee80fd58af5bf5ae2ccb770fdd Mon Sep 17 00:00:00 2001
From: Raphael Townshend <177576+raphtown@users.noreply.github.com>
Date: Mon, 1 Feb 2021 22:29:24 -0800
Subject: [PATCH] Bugfix make_dataset

---
 atom3d/datasets/__main__.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/atom3d/datasets/__main__.py b/atom3d/datasets/__main__.py
index 68a40b7..8c7fdd7 100644
--- 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__":