commit a8c003909f6cfdbe7d71d40196920dd9e01c2467
author James Peet <james.peet@arm.com> Thu Jul 15 13:38:57 2021 +0100
committer James Peet <james.peet@arm.com> Thu Jul 15 17:09:34 2021 +0100
tree 4d68e4b783b2a3d24100e09e147c5133862c6740
parent ffe8e288c8321dfc55a3b75f1aedc08769ecb23a

MLBEDSW-4920 : ModuleNotFoundError for tosa

Adds tosa package to setup.py

Change-Id: I931301313b6d79402e3caf53df7718588a5f538d
Signed-off-by: James Peet <james.peet@arm.com>

setup.py

diff --git a/setup.py b/setup.py
index 9133a12..d4bc009 100644
--- a/setup.py
+++ b/setup.py
@@ -81,7 +81,13 @@
         "Topic :: Software Development :: Compilers",
     ],
     keywords=["ethos-u", "vela compiler", "tflite", "npu"],
-    packages=["ethosu.vela", "ethosu.vela.ethos_u55_regs", "ethosu.vela.tflite", "ethosu.mlw_codec"],
+    packages=[
+        "ethosu.vela",
+        "ethosu.vela.ethos_u55_regs",
+        "ethosu.vela.tflite",
+        "ethosu.vela.tosa",
+        "ethosu.mlw_codec",
+    ],
     python_requires="~=3.6",
     install_requires=[
         "flatbuffers==1.12.0",