グーグルコラボで今まで動いていたプログラムが、動かなくなりました

前提

ここに質問の内容を詳しく書いてください。
画像を2群に判別するプログラムを書いています

実現したいこと

最近まで動いていたのに
給に動かなくなりました

発生している問題・エラーメッセージ

---

AttributeError Traceback (most recent call last)
<ipython-input-9-689a7ef2904d> in <module>
54
55 prediction_loader = torch.utils.data.DataLoader(prediction_subset, batch_size=batch_size, shuffle=True)
---> 56 inputs, _ = iter(prediction_loader).next()
57
58 #inputs, _ = iter(prediction_loader).next(iter(dataloader))

AttributeError: '_SingleProcessDataLoaderIter' object has no attribute 'next'

該当のソースコード

prediction_loader = torch.utils.data.DataLoader(prediction_subset, batch_size=batch_size, shuffle=True)
inputs, _ = iter(prediction_loader).next(iter(dataloader))
prediction(model, inputs)

補足情報（FW/ツールのバージョンなど）

google コラボラトリを用いています