[PATCH] kunit: tool: fix type annotation for IO