Fix pyright errors

pulp-platform · Dec 12, 2024 · bbd4829 · bbd4829
1 parent 40154ca
commit bbd4829
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/test.py b/test/test.py
@@ -107,25 +107,25 @@ def assert_message(msg: str, test_name: str, stdout: str, stderr: Optional[str]
     return retval
 
 
-def build(nnxName: str, flow: Literal["make", "cmake"]) -> None:
+def build(nnxName: NnxName, flow: Literal["make", "cmake"]) -> None:
     env = os.environ
 
     if flow == "make":
         Path("app/src/nnx_layer.c").touch()
         cmd = "make -C app all platform=gvsoc"
-        env["ACCELERATOR"] = nnxName
+        env["ACCELERATOR"] = str(nnxName)
     elif flow == "cmake":
         cmd = "cmake --build app/build"
 
     subprocess.run(cmd.split(), check=True, capture_output=True, text=True, env=env)
 
 
-def run(nnxName: str, flow: Literal["make", "cmake"]) -> str:
+def run(nnxName: NnxName, flow: Literal["make", "cmake"]) -> str:
     env = os.environ
 
     if flow == "make":
         cmd = "make -C app run platform=gvsoc"
-        env["ACCELERATOR"] = nnxName
+        env["ACCELERATOR"] = str(nnxName)
     elif flow == "cmake":
         build_dir = os.path.abspath(f"app/build_{nnxName}")
         bin = os.path.join(build_dir, "test-pulp-nnx")