From cc619775c9153b770dcfa6c5c0fe385a6da81086 Mon Sep 17 00:00:00 2001
From: Nirvedh Meshram <96096277+nirvedhmeshram@users.noreply.github.com>
Date: Sun, 4 Feb 2024 17:39:15 -0600
Subject: [PATCH] Fix mlir printer arguments to match new signatures (#392)

The function signatures seem to have changed after
https://github.com/llvm/llvm-project/pull/75099
---
 python/shark_turbine/aot/exporter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)