fix typo of py args (openxla#2104)

because that the lambda's signature is `[](MlirContext context, std::string artifact)`.
ghpvnist · Mar 18, 2024 · 9c1bccf · 9c1bccf
1 parent b8a10b8
commit 9c1bccf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stablehlo/integrations/python/StablehloModule.cpp b/stablehlo/integrations/python/StablehloModule.cpp
@@ -551,5 +551,5 @@ PYBIND11_MODULE(_stablehlo, m) {
 
         return {module.release()};
       },
-      py::arg("module"), py::arg("target"));
+      py::arg("context"), py::arg("artifact"));
 }