def wrap_if_nodes_inside(raw_array, slow_op_name=None): if raw_array.dtype is np.dtype("O"): if slow_op_name: warnings.warn("{0} is slow for array inputs. " "np.concatenate() is faster.".format(slow_op_name)) return array_from_args(*raw_array.ravel()).reshape(raw_array.shape) else: return raw_array
def wrap_if_nodes_inside(raw_array, slow_op_name=None): if raw_array.dtype is np.dtype('O'): if slow_op_name: warnings.warn("{0} is slow for array inputs. " "np.concatenate() is faster.".format(slow_op_name)) return array_from_args(*raw_array.ravel()).reshape(raw_array.shape) else: return raw_array