diff options
Diffstat (limited to 'ethosu/vela/errors.py')
-rw-r--r-- | ethosu/vela/errors.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/ethosu/vela/errors.py b/ethosu/vela/errors.py index 2c93fbc6..59740aac 100644 --- a/ethosu/vela/errors.py +++ b/ethosu/vela/errors.py @@ -15,6 +15,7 @@ # limitations under the License. # Description: # Defines custom exceptions. +import inspect import sys from .operation import Operation @@ -121,3 +122,20 @@ def TensorError(tens, msg): print("Error: {}".format(data)) sys.exit(1) + + +def typecheck(func): + def wrapper(*args, **kwargs): + fsig = inspect.signature(func) + args_zipped = zip(kwargs.values(), fsig.parameters.keys()) + for actual, expected in args_zipped: + expected_type = fsig.parameters[expected].annotation + actual_type = type(actual) + if expected_type is inspect.Parameter.empty: + raise TypeError("Please provide type info for {}, hint = {}".format(expected, actual_type)) + if expected_type is not actual_type: + raise TypeError("expected : {}, but got {}".format(expected_type, actual_type)) + # Actual execution + return func(*args, **kwargs) + + return wrapper |