diff --git a/python/shark_turbine/aot/exporter.py b/python/shark_turbine/aot/exporter.py index e8c429583..e94e88eca 100644 --- a/python/shark_turbine/aot/exporter.py +++ b/python/shark_turbine/aot/exporter.py @@ -79,7 +79,7 @@ def save_mlir(self, file_path: Union[str, Path]): if file_path.suffix == ".mlirbc": self.mlir_module.write_bytecode(f) else: - self.mlir_module.print(f, binary=True) + self.mlir_module.print(file=f, binary=True) def _run_import(self): CompiledModule.run_import(self.compiled_module)