update download for sk model so catches any exception

kubeflow · Nov 5, 2018 · 355dd23 · 355dd23
1 parent ec4a2da
commit 355dd23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/sk_mnist/train/create_model.py b/models/sk_mnist/train/create_model.py
@@ -8,7 +8,7 @@
 if __name__ == '__main__':
     try:
         mnist = fetch_mldata('MNIST original')
-    except urllib.error.HTTPError as ex:
+    except:
         print("Could not download MNIST data from mldata.org, trying alternative...")
 
         # Alternative method to load MNIST, if mldata.org is down