diff options
| author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-01 12:39:55 +0100 |
|---|---|---|
| committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-21 11:09:39 +0100 |
| commit | 845e23200d471e44f274940846e400d170b5ff37 (patch) | |
| tree | 28a01492bf11f0ff69309ead9bd8a1bad9e14cbb /ethosu/vela/high_level_command_stream.py | |
| parent | d2b5510697e7789f5a416f9d80d3cb640eecc092 (diff) | |
| download | ethos-u-vela-845e23200d471e44f274940846e400d170b5ff37.tar.gz | |
MLBEDSW-3367 Add mypy to pre-commit
Add mypy to pre-commit and clean up all reported errors.
Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com>
Change-Id: If7dc869f5fecdb0e2db40f14e7d9db21aa33df71
Diffstat (limited to 'ethosu/vela/high_level_command_stream.py')
| -rw-r--r-- | ethosu/vela/high_level_command_stream.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/high_level_command_stream.py b/ethosu/vela/high_level_command_stream.py index 7e60221d..0009f6cf 100644 --- a/ethosu/vela/high_level_command_stream.py +++ b/ethosu/vela/high_level_command_stream.py @@ -16,6 +16,7 @@ # Description: # Contains classes that hold commands for the high-level command stream (one command per DMA or NPU stripe). from typing import List +from typing import Optional import numpy as np @@ -41,8 +42,8 @@ class Box: npu_block_type: NpuBlockType, concat_offsets: List[int], k_dilated_height: int, - split_offset: Shape4D = None, - split_shape: Shape4D = None, + split_offset: Optional[Shape4D] = None, + split_shape: Optional[Shape4D] = None, upscaling_factor: int = 1, ): new_start_coord = list(self.start_coord) |