TypeError: predict() got an unexpected keyword argument 'multi_class_prob'

[bluusun]

After upgrading to Python 3.8.10 I got the above error when running the test data as per the install instructions.

Other (likely) new versions:

torch 1.10.1
name2nat 0.5.1
scikit-learn 1.0.2

Same issue on a fresh install.