Fix python binding

reverendbedford · Sep 12, 2024 · 28e8648 · 28e8648
1 parent 9cd390d
commit 28e8648
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/python/neml2/base.cxx b/python/neml2/base.cxx
@@ -309,10 +309,11 @@ tensors.TensorValue.set_ instead to modify the tensor value.
           },
           py::return_value_policy::reference,
           "Get the sub-models registered to this model")
-      .def("get_parameter",
-           &Model::get_parameter,
-           py::return_value_policy::reference,
-           "Get a model parameter given its name")
+      .def(
+          "get_parameter",
+          [](Model & self, const std::string & name) { return &self.get_parameter(name); },
+          py::return_value_policy::reference,
+          "Get a model parameter given its name")
       .def("set_parameter", &Model::set_parameter, "Set the value for a model parameter")
       .def("set_parameters", &Model::set_parameters, "Set the values for multiple model parameters")
       .def(