py: fix AtomicReq bindings

1 files changed, 3 insertions, 3 deletions

diff --git a/py/pykmsbase.cpp b/py/pykmsbase.cpp
index 83be9d2..f31986a 100644
--- a/py/pykmsbase.cpp
+++ b/py/pykmsbase.cpp
@@ -120,8 +120,8 @@ void init_pykmsbase(py::module &m)
 			.def(py::init<Card&>(),
 			     py::keep_alive<1, 2>())	// Keep Card alive until this is destructed
 			.def("add", (void (AtomicReq::*)(DrmObject*, const string&, uint64_t)) &AtomicReq::add)
-			.def("test", &AtomicReq::test)
-			.def("commit", &AtomicReq::commit)
-			.def("commit_sync", &AtomicReq::commit_sync)
+			.def("test", &AtomicReq::test, py::arg("allow_modeset") = false)
+			.def("commit", &AtomicReq::commit, py::arg("data"), py::arg("allow_modeset") = false)
+			.def("commit_sync", &AtomicReq::commit_sync, py::arg("allow_modeset") = false)
 			;
 }