Merge pull request PaddlePaddle#17 from Liwb5/develop

fix pybind of FeatureNode
inpiredhss · Mar 28, 2021 · 2cb2eaf · 2cb2eaf
2 parents 9a47d1d + 8826468
commit 2cb2eaf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/paddle/fluid/pybind/fleet_py.cc b/paddle/fluid/pybind/fleet_py.cc
@@ -166,7 +166,10 @@ void BindGraphNode(py::module* m) {
       .def("get_feature", &GraphNode::get_feature);
 }
 void BindGraphPyFeatureNode(py::module* m) {
-  py::class_<FeatureNode>(*m, "FeatureNode").def(py::init<>());
+  py::class_<FeatureNode>(*m, "FeatureNode")
+      .def(py::init<>())
+      .def("get_id", &GraphNode::get_id)
+      .def("get_feature", &GraphNode::get_feature);
 }
 
 void BindGraphPyService(py::module* m) {