diff --git a/tests/micro/zephyr/test_zephyr.py b/tests/micro/zephyr/test_zephyr.py index a1a1805d9acf1..14e250c3a44ec 100644 --- a/tests/micro/zephyr/test_zephyr.py +++ b/tests/micro/zephyr/test_zephyr.py @@ -210,9 +210,7 @@ def test_relay(temp_dir, platform, west_cmd, tvm_debug): with tvm.transform.PassContext(opt_level=3, config={"tir.disable_vectorize": True}): mod = tvm.relay.build(func, target=target) - with _make_session( - temp_dir, zephyr_board, west_cmd, mod, build_config - ) as session: + with _make_session(temp_dir, zephyr_board, west_cmd, mod, build_config) as session: graph_mod = tvm.micro.create_local_graph_executor( mod.get_graph_json(), session.get_system_lib(), session.device ) @@ -255,9 +253,7 @@ def test_onnx(temp_dir, platform, west_cmd, tvm_debug): lowered = relay.build(relay_mod, target, params=params) graph = lowered.get_graph_json() - with _make_session( - temp_dir, zephyr_board, west_cmd, lowered, build_config - ) as session: + with _make_session(temp_dir, zephyr_board, west_cmd, lowered, build_config) as session: graph_mod = tvm.micro.create_local_graph_executor( graph, session.get_system_lib(), session.device ) @@ -284,9 +280,7 @@ def check_result( with tvm.transform.PassContext(opt_level=3, config={"tir.disable_vectorize": True}): mod = tvm.relay.build(relay_mod, target=target) - with _make_session( - temp_dir, zephyr_board, west_cmd, mod, build_config - ) as session: + with _make_session(temp_dir, zephyr_board, west_cmd, mod, build_config) as session: rt_mod = tvm.micro.create_local_graph_executor( mod.get_graph_json(), session.get_system_lib(), session.device ) diff --git a/tests/micro/zephyr/test_zephyr_aot.py b/tests/micro/zephyr/test_zephyr_aot.py index 1d08a3c5f916d..06250253dee70 100644 --- a/tests/micro/zephyr/test_zephyr_aot.py +++ b/tests/micro/zephyr/test_zephyr_aot.py @@ -47,9 +47,7 @@ PLATFORMS = conftest.PLATFORMS -def _build_project( - temp_dir, zephyr_board, west_cmd, mod, build_config, extra_files_tar=None -): +def _build_project(temp_dir, zephyr_board, west_cmd, mod, build_config, extra_files_tar=None): template_project_dir = ( pathlib.Path(__file__).parent / ".."