|
|
|
@@ -390,8 +390,8 @@ Status BatchOp::InvokeBatchMapFunc(TensorTable *input, TensorTable *output, CBat |
|
|
|
TensorRow output_batch; |
|
|
|
// If user returns a type that is neither a list nor an array, issue a error msg. |
|
|
|
if (!py::isinstance<py::list>(ret_tuple[i])) { |
|
|
|
MS_LOG(ERROR) << "column: " << out_col_names_[i] |
|
|
|
<< " returned by per_batch_map is not a list, this could lead to conversion failure."; |
|
|
|
MS_LOG(INFO) << "column: " << out_col_names_[i] |
|
|
|
<< " returned by per_batch_map is not a list, this could lead to conversion failure."; |
|
|
|
} |
|
|
|
|
|
|
|
py::list output_list = py::cast<py::list>(ret_tuple[i]); |
|
|
|
|