Training phase problem

[shadow1999k]

Hi, As I was trying to run README_CPO-ml-100k.ipynb, in the last line of the code for 'Train' cell which is : cpo.train(n_episodes, modelPath)

I got the error : forward() takes 2 positional arguments but 3 were given.

I'm using torch==1.10.2 and data prepration was done as in run_data_preprocess.ipynb without any problems.

Any idea what might the reason be? Looking forward to your answer, thank you.

[shadow1999k]

@TobyGE @YingqiangGe