Skip to content

Commit f386e00

Browse files
authored
Move comm (#585)
* Move comm package * Update the comm module * Remove leftover
1 parent ca9a67c commit f386e00

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

src/xcomm.cpp

+10
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,16 @@ namespace xpyt
144144
.def(py::init<>())
145145
.def("register_target", &xcomm_manager::register_target);
146146

147+
comm_module.def("create_comm", [&comm_module](py::args objs, py::kwargs kw) {
148+
return comm_module.attr("Comm")(*objs, **kw);
149+
});
150+
151+
comm_module.def("get_comm_manager", [&comm_module]() {
152+
static py::object comm_manager = comm_module.attr("CommManager")();
153+
154+
return comm_manager;
155+
});
156+
147157
return comm_module;
148158
}
149159

src/xinterpreter.cpp

+3-1
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,10 @@ namespace xpyt
7373
py::module comm_module = get_comm_module();
7474
py::module kernel_module = get_kernel_module();
7575

76-
// Monkey patching "from ipykernel.comm import Comm"
76+
// Old approach: ipykernel provides the comm
7777
sys.attr("modules")["ipykernel.comm"] = comm_module;
78+
// New approach: we provide our comm module
79+
sys.attr("modules")["comm"] = comm_module;
7880

7981
instanciate_ipython_shell();
8082

0 commit comments

Comments
 (0)