add dataset name to download.py
This commit is contained in:
@@ -11,7 +11,7 @@ curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/get-poet
|
|||||||
```
|
```
|
||||||
source load-env.sh
|
source load-env.sh
|
||||||
poetry run python list.py
|
poetry run python list.py
|
||||||
poetry run python download.py
|
poetry run python download.py <dataset name>
|
||||||
```
|
```
|
||||||
|
|
||||||
## how this was done
|
## how this was done
|
||||||
|
@@ -4,7 +4,12 @@ import sys
|
|||||||
|
|
||||||
import datasets
|
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))
|
print(len(mats))
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user