RuntimeError: as_numpy_iterator() is not supported while tracing functions on tensorflow 2.x dataset on sagemaker

Question

I am preparing a sagemaker PIPE mode dataset to train a time series model on SageMaker with PIPE mode. The PipeModeDataset is a