deprecated 가 의심되는 코드 하나 더 여쭙겠습니다

예제 22 IMDB_RNN.py  에서

train_dataset = train_dataset.shuffle(BUFFER_SIZE)

train_dataset = train_dataset.padded_batch(BATCH_SIZE,  train_dataset.output_shapes))

test_dataset = test_dataset.padded_batch(BATCH_SIZE,  test_dataset.output_shapes)

이 부분에

train_dataset.output_shapes  와 test_dataset.output_shapes 가 에러입니다

에러 메세지는 다음과 같습니다
'ShuffleDataset' object has no attribute 'output_shapes'

이부분... 구글 colab 에서는 약간의 경고 메세지와 함께 실행은 되는데
주피터 노트북에서는 위와 같은 에러가 납니다

죄송하지만 이거 하나더 부탁드리겠습니다