Skip to content

Commit 0566df6

Browse files
[TRTLLM-6707][fix] nanobind fix for executor exit call (#7565)
Signed-off-by: Linda-Stadter <[email protected]>
1 parent dcd110c commit 0566df6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cpp/tensorrt_llm/nanobind/executor/executor.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -199,7 +199,7 @@ void Executor::initBindings(nb::module_& m)
199199
nb::arg("decoder_json_config_str"), nb::arg("model_type"), nb::arg("executor_config"))
200200
.def("shutdown", &Executor::shutdown)
201201
.def("__enter__", &Executor::enter)
202-
.def("__exit__", &Executor::exit)
202+
.def("__exit__", &Executor::exit, nb::arg("type").none(), nb::arg("value").none(), nb::arg("traceback").none())
203203
.def("enqueue_request", &Executor::enqueueRequest, nb::arg("request"))
204204
.def("enqueue_requests", &Executor::enqueueRequests, nb::arg("requests"))
205205
.def("await_responses",

0 commit comments

Comments
 (0)