Tensorflow py_function setting return value shape

Question

I am tring to set the shape of an numpy array return from a tf.py_function call. Function returns a tuple of array and its shape, so I can set its shape inside