From 291ae1a1ee757dca7c4b7f853588fda24f28c3ae Mon Sep 17 00:00:00 2001
From: Carl Pearson <cwpears@sandia.gov>
Date: Tue, 23 Nov 2021 07:48:43 -0800
Subject: [PATCH] add dataset name to download.py

---
 README.md   | 2 +-
 download.py | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 1abff8b..22809a2 100644
--- a/README.md
+++ b/README.md
@@ -11,7 +11,7 @@ curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poet
 ```
 source load-env.sh
 poetry run python list.py
-poetry run python download.py
+poetry run python download.py <dataset name>
 ```
 
 ## how this was done
diff --git a/download.py b/download.py
index c67d9a2..12c14f2 100755
--- a/download.py
+++ b/download.py
@@ -4,7 +4,12 @@ import sys
 
 import datasets
 
-mats = datasets.ALL_REAL_MATS
+dataset = None
+for ds in datasets.DATASETS:
+    if ds.name == sys.argv[1]:
+        dataset = ds
+        break
+mats = dataset.mats
 
 print(len(mats))