diff --git a/pymodule/pycatima.cpp b/pymodule/pycatima.cpp index 2759488..539f30b 100644 --- a/pymodule/pycatima.cpp +++ b/pymodule/pycatima.cpp @@ -146,7 +146,8 @@ PYBIND11_MODULE(pycatima,m){ py::class_(m,"Layers") .def(py::init<>(),"constructor") - .def("add",&Layers::add) + .def("add",py::overload_cast(&Layers::add)) + .def("add_layers",py::overload_cast(&Layers::add)) .def("num",&Layers::num) // .def("__getitem__",&Layers::operator[], py::is_operator()) .def("__getitem__",[](Layers &r, int i)->Material*