small fix

Hongshan Li · Hongshan Li · commit 8b8686f09dfd · 2021-03-09T19:51:29.000Z
diff --git a/utils/datasets.py b/utils/datasets.py
@@ -1,4 +1,5 @@
 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()

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`import boto3`
	`2`	`+import os`
`2`	`3`
`3`	`4`	`PUBLIC_BUCKET = "sagemaker-sample-files"`
`4`	`5`