I encountered exactly the same problem, if the array has 3 dimensions set_format fails and I obtain list of 2d tensors instead of one 3d tensor. To solve this problem you have to simply cast column to Array3D type with, e.g., Dataset.cast_column function