KeyError: 'length' when using using load_dataset on Sagemaker

It appears in the notebook. I have not started the sagemaker training job and this error appears. Thanks!