hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix bug in numbuf serialization (#296)

Browse source
This commit is contained in:
Philipp Moritz 2017-02-17 23:35:41 -08:00 • committed by Robert Nishihara
parent a0dd3a44c0
commit 9973a6e37c
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/numbuf/python/src/pynumbuf/adapters/python.cc
View file

@ -98,13 +98,14 @@ Status append(PyObject* elem, SequenceBuilder& builder, std::vector<PyObject*>&
Py_ssize_t size;
#if PY_MAJOR_VERSION >= 3
char* data = PyUnicode_AsUTF8AndSize(elem, &size);
Status s = builder.AppendString(data, size);
#else
PyObject* str = PyUnicode_AsUTF8String(elem);
char* data = PyString_AS_STRING(str);
size = PyString_GET_SIZE(str);
Status s = builder.AppendString(data, size);
Py_XDECREF(str);
#endif
Status s = builder.AppendString(data, size);
RETURN_NOT_OK(s);
} else if (PyList_Check(elem)) {
builder.AppendList(PyList_Size(elem));