mirrors/sby
3
0
Fork 0
mirror of https://github.com/YosysHQ/sby.git synced 2025-08-08 06:11:24 +00:00
Code Activity

tools/cexenum: Fix type annotations

Browse source 
With this it now passes pyright's strict mode
This commit is contained in:
Jannis Harder 2023-11-29 13:41:16 +01:00
parent f0f140c83c
commit aa2a96d7c9
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tools/cexenum/cexenum.py
View file

@ -83,7 +83,7 @@ def arg_parser():
return parser return parser
def lines(*args): def lines(*args: Any):
return "".join(f"{line}\n" for line in args) return "".join(f"{line}\n" for line in args)
@ -148,7 +148,7 @@ def setup_logging():
tl.current_task().set_error_handler(None, error_handler) tl.current_task().set_error_handler(None, error_handler)
async def batch(*args): async def batch(*args: Awaitable[Any]):
result = None result = None
for arg in args: for arg in args:
result = await arg result = await arg
@ -200,7 +200,7 @@ class AigModel(tl.process.Process):
"hierarchy -simcheck", "hierarchy -simcheck",
"flatten", "flatten",
"setundef -undriven -anyseq", "setundef -undriven -anyseq",
"setattr -set keep 1 w:\*", "setattr -set keep 1 w:\\*",
"delete -output", "delete -output",
"opt -full", "opt -full",
"techmap", "techmap",
@ -299,7 +299,7 @@ def relative_to(target: Path, cwd: Path) -> Path:
class YosysWitness(tl.process.Process): class YosysWitness(tl.process.Process):
def __init__( def __init__(
self, self,
mode: Literal["yw2aiw"] | Literal["aiw2yw"], mode: Literal["yw2aiw", "aiw2yw"],
input: Path, input: Path,
mapfile: Path, mapfile: Path,
output: Path, output: Path,
@ -477,10 +477,10 @@ class Smtbmc(tl.process.Process):
) )
self.name = "smtbmc" self.name = "smtbmc"
self.expected_results = [] self.expected_results: list[asyncio.Future[Any]] = []
async def on_run(self) -> None: async def on_run(self) -> None:
def output_handler(event: tl.process.StderrEvent): def output_handler(event: tl.process.StdoutEvent):
result = json.loads(event.output) result = json.loads(event.output)
tl.log_debug(f"smtbmc > {result!r}") tl.log_debug(f"smtbmc > {result!r}")
if "err" in result: if "err" in result:
@ -501,11 +501,11 @@ class Smtbmc(tl.process.Process):
def ping(self) -> Awaitable[None]: def ping(self) -> Awaitable[None]:
return self.incremental_command(cmd="ping") return self.incremental_command(cmd="ping")
def incremental_command(self, **command: dict[Any]) -> Awaitable[Any]: def incremental_command(self, **command: Any) -> Awaitable[Any]:
tl.log_debug(f"smtbmc < {command!r}") tl.log_debug(f"smtbmc < {command!r}")
self.write(json.dumps(command)) self.write(json.dumps(command))
self.write("\n") self.write("\n")
result = asyncio.Future() result: asyncio.Future[Any] = asyncio.Future()
self.expected_results.append(result) self.expected_results.append(result)
return result return result
@ -565,7 +565,7 @@ class Smtbmc(tl.process.Process):
assertions: bool | None = True, assertions: bool | None = True,
pred: int | None = None, pred: int | None = None,
) -> Awaitable[None]: ) -> Awaitable[None]:
futures = [] futures: list[Awaitable[None]] = []
futures.append(self.new_step(step)) futures.append(self.new_step(step))
futures.append(self.hierarchy(step)) futures.append(self.hierarchy(step))
futures.append(self.assumptions(step)) futures.append(self.assumptions(step))