Add missing macro

This commit is contained in:
Eric 2020-09-24 22:39:08 -04:00
parent a93476ba5d
commit 5a0a35ed47
1 changed files with 1 additions and 1 deletions

View File

@ -69,7 +69,7 @@ PYBIND_REGISTER(Tensor, 0, ([](const py::module *m) {
auto &tensor = py::cast<Tensor &>(t);
if (tensor.type() == DataType::DE_STRING) {
py::array res;
tensor.GetDataAsNumpyStrings(&res);
THROW_IF_ERROR(tensor.GetDataAsNumpyStrings(&res));
return res;
}
py::buffer_info info;