Mnist download (#2064)

hongshanli23 · web-flow · commit f17c40f8be10 · 2021-03-09T11:54:21.000-08:00
* helper fn to download mnist from public s3

* small fix
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -1,4 +1,6 @@
 import boto3
+import os
+
 
 PUBLIC_BUCKET = "sagemaker-sample-files"
 
diff --git a/utils/tests.py b/utils/tests.py
@@ -0,0 +1,3 @@
+import datasets
+
+datasets.download_mnist()
