added errors handling and updated req files

Paperspace · May 14, 2019 · e9819e4 · e9819e4
1 parent 51b5608
commit e9819e4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/mnist.py b/mnist.py
@@ -22,6 +22,7 @@
 from absl import app as absl_app
 from absl import flags
 import tensorflow as tf  # pylint: disable=g-bad-import-order
+from gradient_sdk.exceptions import ConfigError
 
 gradient_sdk = True
 try:
@@ -258,7 +259,7 @@ def main(_):
     if gradient_sdk:
         try:
             get_tf_config()
-        except:
+        except ConfigError:
             pass
     define_mnist_flags()
     # Print ENV Variables

diff --git a/requirements-local.txt b/requirements-local.txt
@@ -0,0 +1,3 @@
+-r requirements.txt
+absl-py
+tensorflow==1.13.1
diff --git a/requirements.txt b/requirements.txt
@@ -12,6 +12,4 @@ typing
 matplotlib
 gradient-sdk
 pillow
-requests
-absl-py # for local run
-tensorflow==1.13.1 # for local run
+requests