diff --git a/python/csrc/port_channel_py.cpp b/python/csrc/port_channel_py.cpp index 0cda37352..4b1aa2892 100644 --- a/python/csrc/port_channel_py.cpp +++ b/python/csrc/port_channel_py.cpp @@ -12,12 +12,12 @@ using namespace mscclpp; void register_port_channel(nb::module_& m) { nb::class_(m, "BaseProxyService") - .def("start_proxy", &BaseProxyService::startProxy) + .def("start_proxy", &BaseProxyService::startProxy, nb::arg("blocking") = false) .def("stop_proxy", &BaseProxyService::stopProxy); nb::class_(m, "ProxyService") .def(nb::init(), nb::arg("fifo_size") = DEFAULT_FIFO_SIZE) - .def("start_proxy", &ProxyService::startProxy) + .def("start_proxy", &ProxyService::startProxy, nb::arg("blocking") = false) .def("stop_proxy", &ProxyService::stopProxy) .def("build_and_add_semaphore", &ProxyService::buildAndAddSemaphore, nb::arg("comm"), nb::arg("connection")) .def("add_semaphore", static_cast(&ProxyService::addSemaphore),