From b7be390fd3521967979f452c5cafede49d6925c6 Mon Sep 17 00:00:00 2001 From: Guru Deshpande <66385690+guru-desh@users.noreply.github.com> Date: Mon, 13 Jul 2026 20:23:45 -0700 Subject: [PATCH 1/2] fix(test): change AIModel asset saving to give temp path to AIModel instead of directory --- tests/export/export_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/export/export_utils.py b/tests/export/export_utils.py index 8704db7..b27b94f 100644 --- a/tests/export/export_utils.py +++ b/tests/export/export_utils.py @@ -459,7 +459,8 @@ async def _run_inference_async( prefix="mlir_converter_inference", suffix=".aimodel", ) as tmpdir: - asset = converted_model.save_asset(Path(tmpdir)) + asset_path = Path(tmpdir) / "asset.aimodel" + asset = converted_model.save_asset(asset_path) async with asset.executable( specialization_options=_get_test_specialization_options(), ) as ai_model: From fe0310a76419235941bf587e8f3570291c8e0ccb Mon Sep 17 00:00:00 2001 From: Guru Deshpande <66385690+guru-desh@users.noreply.github.com> Date: Tue, 14 Jul 2026 00:32:52 -0700 Subject: [PATCH 2/2] fix(test): drop vestigial .aimodel suffix from inference tempdir --- tests/export/export_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/export/export_utils.py b/tests/export/export_utils.py index b27b94f..9553496 100644 --- a/tests/export/export_utils.py +++ b/tests/export/export_utils.py @@ -457,7 +457,6 @@ async def _run_inference_async( """Async implementation of MLIR inference.""" with tempfile.TemporaryDirectory( prefix="mlir_converter_inference", - suffix=".aimodel", ) as tmpdir: asset_path = Path(tmpdir) / "asset.aimodel" asset = converted_model.save_asset(asset_path)