recover_save_act_in_fit() missing 1 required positional argument: 'old_save_act'

[liuyh-Horizon]

Hello, I'm a beginner to use KAN. I tried to use the model.fit() to train a small network for MNIST. And I got one error like this:

TypeError: recover_save_act_in_fit() missing 1 required positional argument: 'old_save_act'

Can anyone help me to find out what happened here? I will enclose my codes:

Thanks a lot!!