add dataset name to download.py

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

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))