diff options
Diffstat (limited to 'python/pyarmnn/test/test_generated.py')
-rw-r--r-- | python/pyarmnn/test/test_generated.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/python/pyarmnn/test/test_generated.py b/python/pyarmnn/test/test_generated.py index f27d565c2b..60d686afcb 100644 --- a/python/pyarmnn/test/test_generated.py +++ b/python/pyarmnn/test/test_generated.py @@ -9,7 +9,6 @@ import pyarmnn._generated.pyarmnn as generated_armnn import pyarmnn._generated.pyarmnn as generated_deserializer import pyarmnn._generated.pyarmnn_onnxparser as generated_onnx import pyarmnn._generated.pyarmnn_tfliteparser as generated_tflite -import pyarmnn._generated.pyarmnn_tfparser as generated_tf swig_independent_classes = ('IBackend', 'IDeviceSpec', @@ -28,8 +27,7 @@ def get_classes(swig_independent_classes: Tuple): inspect.getmembers(generated_armnn, inspect.isclass) + inspect.getmembers(generated_deserializer, inspect.isclass) + inspect.getmembers(generated_tflite, inspect.isclass) + - inspect.getmembers(generated_onnx, inspect.isclass) + - inspect.getmembers(generated_tf, inspect.isclass))) + inspect.getmembers(generated_onnx, inspect.isclass))) @pytest.mark.parametrize("class_instance", get_classes(swig_independent_classes), ids=lambda x: 'class={}'.format(x[0])) |