__getitem__() takes 1 positional argument but 2 were givenこのエラーの解決方法が分かる方教えていただきたいです。

前提

下記に添付したコードを入力したのですが、引数を１つにしていますが、２つあるというエラーが生じてしまいました。

実現したいこと

下記のコードを通したいです。

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

Input In [19], in <cell line: 3>() 1 #トレーニングデータセットの確認 2 index = 0 ----> 3 print(train_dataset.__getitem__(index)[0].shape) 4 print(train_dataset.__getitem__(index)[1]) TypeError: __getitem__() takes 1 positional argument but 2 were given

該当のソースコード

pytorch

#トレーニングデータセットの確認 index = 0 print(train_dataset.__getitem__(index)[0].shape) print(train_dataset.__getitem__(index)[1])

試したこと

引数を１つに変更したのですが上記のまま変わらずでした。

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

python3.8.10
ご協力お願いいたします。