diff options
author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 10:30:25 +0200 |
---|---|---|
committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 15:54:14 +0200 |
commit | d85750702229af97c0b0bbda6e397a23254b6144 (patch) | |
tree | 389962105a35d5cef595cfeb5d640bd59a0d0ff8 /ethosu/vela/test/extapi/test_extapi_encode_weights.py | |
parent | cc5f4de1c35ba44fca7ff6295c6ae846f8242344 (diff) | |
download | ethos-u-vela-d85750702229af97c0b0bbda6e397a23254b6144.tar.gz |
Update version of Black to 22.3.0
Update version of Black to 22.3.0 due to updated dependencies.
Updates to fix reported issues due to new version.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: I60056aae452093ce8dcea1f499ecced22b25eef1
Diffstat (limited to 'ethosu/vela/test/extapi/test_extapi_encode_weights.py')
-rw-r--r-- | ethosu/vela/test/extapi/test_extapi_encode_weights.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/ethosu/vela/test/extapi/test_extapi_encode_weights.py b/ethosu/vela/test/extapi/test_extapi_encode_weights.py index 6367cb30..87c504f4 100644 --- a/ethosu/vela/test/extapi/test_extapi_encode_weights.py +++ b/ethosu/vela/test/extapi/test_extapi_encode_weights.py @@ -24,7 +24,8 @@ from ethosu.vela.api import NpuBlockTraversal @pytest.mark.parametrize( - "arch", list(NpuAccelerator), + "arch", + list(NpuAccelerator), ) @pytest.mark.parametrize("dilation_x", [1, 2]) @pytest.mark.parametrize("dilation_y", [1, 2]) @@ -32,7 +33,12 @@ from ethosu.vela.api import NpuBlockTraversal @pytest.mark.parametrize("depth_control", [1, 2, 3]) @pytest.mark.parametrize("weights_shape_and_block_depth", [((16, 16, 16, 16), 8), ((3, 3, 25, 16), 8)]) def test_encode_weights( - arch, weights_shape_and_block_depth, dilation_x, dilation_y, ifm_bitdepth, depth_control, + arch, + weights_shape_and_block_depth, + dilation_x, + dilation_y, + ifm_bitdepth, + depth_control, ): """ This unit test checks the interface of the API function but not the functionality. |