How to solute this problem in dataset.py(ProteinDataset method)

[Planet-KIM]

Traceback (most recent call last):
  File "dataset.py", line 383, in <module>
    test()
  File "dataset.py", line 375, in test
    dataset = ProteinDataset('T1019s2.pkl', DISTOGRAM_MODEL, replica)
TypeError: __init__() takes 3 positional arguments but 4 were given

• Please give me the solution this problem.
• I want to test the dataset.py.