Added error handling for sample inference test in model deployment

Author: Bradley Munday

modzy/__init__.py

@@ -5,6 +5,6 @@
 
 from .client import ApiClient  # noqa
 from .edge.client import EdgeClient
-__version__ = '0.10.1'
+__version__ = '0.10.2'
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())

modzy/_util.py

@@ -150,6 +150,9 @@ def run_model(client, logger, identifier, version):
         time.sleep(1)
 
     test_output = res['result']
+    # perform validation check on test_output and raise error if error exists
+    if test_output["status"] == "FAILED":
+        raise ValueError(f'Sample inference test failed with error {test_output["error"]}. Check model container and try again.')
 
     sample_input = {'input': {'accessKeyID': '<accessKeyID>',
                                     'region': '<region>',

setup.py

@@ -40,6 +40,6 @@
     # removed in 0.7.1 test_suite='tests',
     # removed in 0.7.1 tests_require=test_requirements,
     url='https://github.com/modzy/sdk-python',
-    version='0.10.1',
+    version='0.10.2',
     zip_safe=False,
 )