Change Operator::create use py::bytes not std::string

8 years ago · 0a0b4caaa7
parent d3373c5b85
commit 0a0b4caaa7
1 changed files with 1 additions and 1 deletions
--- a/paddle/pybind/pybind.cc
+++ b/paddle/pybind/pybind.cc
@ -110,7 +110,7 @@ All parameter, weight, gradient are variables in Paddle.
  py::class_<pd::OperatorBase, pd::OperatorPtr>(m, "Operator")
      .def("__str__", &pd::OperatorBase::DebugString)
      .def_static("create",
-                  [](const std::string& protobin) {
+                  [](py::bytes protobin) {
                    pd::OpDesc desc;
                    PADDLE_ENFORCE(desc.ParsePartialFromString(protobin),
                                   "Cannot parse user input to OpDesc");