From 950f28036ca985cf71a0e2a86fda68f1de0adc73 Mon Sep 17 00:00:00 2001 From: Matthew Barrett Date: Mon, 20 Dec 2021 12:22:54 +0000 Subject: [PATCH] Fix dma linting Change-Id: I0aad319b7722a56b5c7f116adcc791951223baa3 --- python/tvm/relay/backend/contrib/ethosu/te/dma.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/backend/contrib/ethosu/te/dma.py b/python/tvm/relay/backend/contrib/ethosu/te/dma.py index b1c0fae88335..14aa67bb37d3 100644 --- a/python/tvm/relay/backend/contrib/ethosu/te/dma.py +++ b/python/tvm/relay/backend/contrib/ethosu/te/dma.py @@ -103,7 +103,11 @@ def read_compute( def write_compute( - tensor: te.Tensor, zero_point: int, scale: float, layout: Optional[str] = None, attrs: dict = None + tensor: te.Tensor, + zero_point: int, + scale: float, + layout: Optional[str] = None, + attrs: dict = None, ) -> te.Tensor: """A tensor expression which represents a write.