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

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.

