tensor_annotations.tensorflow doesn't work with tf.experimental.ExtensionType

[legendof-selda]

I am trying to use tensor_annotations inside a tf.experimental.ExtensionType class.
But it gives the error Unsupported type annotation tensor_annotations.tensorflow.Tensor3[tensor_annotations.tensorflow.uint16, tensor_annotations.axes.Batch, tensor_annotations.axes.Time, tensor_annotations.axes.Features]

Is there any plans to add support for tf.experimental.ExtensionType? Perhaps any workaround?

[mrahtz]

Hey, sorry for the slow reply - it's been a busy couple of weeks.

I've no plans to add support for this any time soon - but if you were wiling to figure it out yourself, I'd happily accept a PR.