Add property for MNIST dataset URL

Author: sutaakar

support/environment.go

@@ -44,6 +44,9 @@ const (
 
 	// Hostname of the Kubernetes cluster
 	ClusterHostname = "CLUSTER_HOSTNAME"
+
+	// URL for downloading MNIST dataset
+	mnistDatasetURL = "MNIST_DATASET_URL"
 )
 
 type ClusterType string
@@ -110,6 +113,10 @@ func GetClusterHostname(t Test) string {
 	return hostname
 }
 
+func GetMnistDatasetURL() string {
+	return lookupEnvOrDefault(mnistDatasetURL, "http://yann.lecun.com/exdb/mnist/")
+}
+
 func lookupEnvOrDefault(key, value string) string {
 	if v, ok := os.LookupEnv(key); ok {
 		return v