Dataset does not support length of feature shape > 1

[akashsharma02]

Hi,

Thank you for this excellent project!
I am trying to create a custom dataset with additional sensory information (such as tactile data) which is an Array3D tensor, but find that when I use the approach mentioned in #547, there is no support to add custom tensor like observations to the episode buffer.

Specifically there are assertions that require the feature shape to be a 1D array at most here