Function run_cli

test/null/test_viz.py:917–922

(*cli_args)

# launch viz cli without subprocess
def run_cli(*cli_args) -> list[dict]:
  from tinygrad.viz.cli import main, get_arg_parser
  args = get_arg_parser().parse_args(cli_args+("--json",))
  with contextlib.redirect_stdout(buf:=io.StringIO()):
    main(args)
  return [json.loads(line) for line in buf.getvalue().strip().splitlines()]

@contextlib.contextmanager
def write_files(viz) -> list[str]: