diff options
Diffstat (limited to 'ethosu/mlw_codec/test/test_mlw_codec.py')
| -rw-r--r-- | ethosu/mlw_codec/test/test_mlw_codec.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ethosu/mlw_codec/test/test_mlw_codec.py b/ethosu/mlw_codec/test/test_mlw_codec.py index 18c828a3..3ff26e53 100644 --- a/ethosu/mlw_codec/test/test_mlw_codec.py +++ b/ethosu/mlw_codec/test/test_mlw_codec.py @@ -15,6 +15,9 @@ # See the License for the specific language governing permissions and # limitations under the License. # Simple example of the usage of mlw_codec. +from typing import Any +from typing import List + import pytest from ethosu import mlw_codec @@ -68,7 +71,7 @@ class TestMLWCodec: with pytest.raises(Exception): mlw_codec.encode(input) - invalid_decode_test_data = [None, 3, []] + invalid_decode_test_data: List[Any] = [None, 3, []] @pytest.mark.parametrize("input", invalid_decode_test_data) def test_decode_invalid_input(self, input): |