fix pybind int64 problem

Author: seemingwang

paddle/fluid/distributed/ps/table/graph/graph_node.h

@@ -20,8 +20,8 @@
 #include <vector>
 #include "glog/logging.h"
 #include "paddle/fluid/distributed/ps/table/graph/graph_weighted_sampler.h"
-#include "paddle/phi/core/enforce.h"
 #include "paddle/fluid/string/string_helper.h"
+#include "paddle/phi/core/enforce.h"
 
 namespace paddle {
 namespace distributed {
@@ -33,6 +33,7 @@ class Node {
   virtual ~Node() {}
   static int id_size, int_size, weight_size;
   uint64_t get_id() { return id; }
+  int64_t get_py_id() { return (int64_t)id; }
   void set_id(uint64_t id) { this->id = id; }
 
   virtual void build_edges(bool is_weighted) {}
@@ -49,7 +50,9 @@ class Node {
   virtual void to_buffer(char *buffer, bool need_feature);
   virtual void recover_from_buffer(char *buffer);
   virtual std::string get_feature(int idx) { return std::string(""); }
-  virtual int get_feature_ids(int slot_idx, std::vector<uint64_t>* res) const { return 0; }
+  virtual int get_feature_ids(int slot_idx, std::vector<uint64_t> *res) const {
+    return 0;
+  }
   virtual void set_feature(int idx, std::string str) {}
   virtual void set_feature_size(int size) {}
   virtual int get_feature_size() { return 0; }
@@ -99,15 +102,15 @@ class FeatureNode : public Node {
     }
   }
 
-  virtual int get_feature_ids(int slot_idx, std::vector<uint64_t>* res) const {
+  virtual int get_feature_ids(int slot_idx, std::vector<uint64_t> *res) const {
     PADDLE_ENFORCE_NOT_NULL(res);
     res->clear();
     errno = 0;
     if (slot_idx < (int)this->feature.size()) {
-      const char* feat_str = this->feature[slot_idx].c_str();
+      const char *feat_str = this->feature[slot_idx].c_str();
       auto fields = paddle::string::split_string<std::string>(feat_str, " ");
-      char* head_ptr = NULL;
-      for (auto &field: fields) {
+      char *head_ptr = NULL;
+      for (auto &field : fields) {
         PADDLE_ENFORCE_EQ(field.empty(), false);
         uint64_t feasign = strtoull(field.c_str(), &head_ptr, 10);
         PADDLE_ENFORCE_EQ(field.c_str() + field.length(), head_ptr);

paddle/fluid/pybind/fleet_py.cc

@@ -178,13 +178,13 @@ void BindHeterClient(py::module* m) {
 void BindGraphNode(py::module* m) {
   py::class_<GraphNode>(*m, "GraphNode")
       .def(py::init<>())
-      .def("get_id", &GraphNode::get_id)
+      .def("get_id", &GraphNode::get_py_id)
       .def("get_feature", &GraphNode::get_feature);
 }
 void BindGraphPyFeatureNode(py::module* m) {
   py::class_<FeatureNode>(*m, "FeatureNode")
       .def(py::init<>())
-      .def("get_id", &GraphNode::get_id)
+      .def("get_id", &GraphNode::get_py_id)
       .def("get_feature", &GraphNode::get_feature);
 }