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