Commit 9512bdaa authored by Dominik Charousset's avatar Dominik Charousset

Fix module initialization on Python 2

parent 0ddf1f29
...@@ -556,7 +556,16 @@ public: ...@@ -556,7 +556,16 @@ public:
} }
}; };
PyObject* caf_module_init() { #if PY_MAJOR_VERSION == 3
#define CAF_MODULE_INIT_RES PyObject*
#define CAF_MODULE_INIT_RET(res) return res;
#else
#define CAF_MODULE_INIT_RES void
#define CAF_MODULE_INIT_RET(unused)
#endif
CAF_MODULE_INIT_RES caf_module_init() {
pybind11::module m("CAF", "Python binding for CAF"); pybind11::module m("CAF", "Python binding for CAF");
s_context->cfg.py_init(m); s_context->cfg.py_init(m);
// add classes // add classes
...@@ -566,7 +575,7 @@ PyObject* caf_module_init() { ...@@ -566,7 +575,7 @@ PyObject* caf_module_init() {
.def("dequeue_message_with_timeout", &py_dequeue_with_timeout, "Receives the next message") .def("dequeue_message_with_timeout", &py_dequeue_with_timeout, "Receives the next message")
.def("self", &py_self, "Returns the global self handle") .def("self", &py_self, "Returns the global self handle")
.def("atom", &atom_from_string, "Creates an atom from a string"); .def("atom", &atom_from_string, "Creates an atom from a string");
return m.ptr(); CAF_MODULE_INIT_RET(m.ptr())
} }
} // namespace python } // namespace python
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment