Fix the lock.png not found issue in model.plot

[Animadversio]

Thanks for the amazing repo and research, super inspiring!

It seems that in the original pykan the lock.png is not included in the distribution. Due to this model.plot(.) could generate file not found error at the line

pykan/kan/KAN.py

Line 647 in 27b4640

im = plt.imread(f'{folder}/lock.png')

I think the lock.png need to be included in the distribution to be load and plotted properly.

Minimal code to reproduce the issue in a Colab notebook

!pip install pykan

from kan import KAN, create_dataset
# initialize KAN with G=3
model = KAN(width=[2,1,1], grid=10, k=3)
# create dataset
f = lambda x: (x[:,[0]]+x[:,[1]])/(1+x[:,[0]]*x[:,[1]])
dataset = create_dataset(f, n_var=2, ranges=[-0.9,0.9])

model.train(dataset, opt="LBFGS", steps=20);
model.plot(beta=10)

model.lock(0,[[0,0],[1,0]])
model.plot(beta=10)

Fix

1. Make the lock.png part of the distribution in setup.py and SOURCES.txt
2. Change the loading path of the lock.png to a package variable RESOURCE_DIR.

The new version passed the test.

!pip install git+https://github.com/Animadversio/pykan.git

from kan import KAN, create_dataset
# initialize KAN with G=3
model = KAN(width=[2,1,1], grid=10, k=3)
# create dataset
f = lambda x: (x[:,[0]]+x[:,[1]])/(1+x[:,[0]]*x[:,[1]])
dataset = create_dataset(f, n_var=2, ranges=[-0.9,0.9])

model.train(dataset, opt="LBFGS", steps=20);
model.plot(beta=10)

model.lock(0,[[0,0],[1,0]])
model.plot(beta=10)