From b914c70c3ac42f959c97781db956015999c254c3 Mon Sep 17 00:00:00 2001
From: Luke Hutton <luke.hutton@arm.com>
Date: Thu, 2 Dec 2021 11:15:01 +0000
Subject: [PATCH] [microNPU] Add RequantizeArgs import to MeanParams

Fixes failing CI since CI on #9576 passed before #9624 was merged.

Change-Id: I44f6aa0abe09d206f3e948e36e9799eaa12b3d14
---
 python/tvm/relay/op/contrib/ethosu.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/tvm/relay/op/contrib/ethosu.py b/python/tvm/relay/op/contrib/ethosu.py
index e53be02b8dd2..3016c3be4ce4 100644
--- a/python/tvm/relay/op/contrib/ethosu.py
+++ b/python/tvm/relay/op/contrib/ethosu.py
@@ -956,6 +956,8 @@ class MeanParams:
     composite_name = "ethos-u.mean"
 
     def __init__(self, func_body: Call):
+        from tvm.relay.backend.contrib.ethosu.util import RequantArgs
+
         requantize = func_body
         mean_op = requantize.args[0]
         attrs = mean_op.attrs