feat: explainer support pdf

[sbrugman]

Enable PDF (and other supported formats) savefig in explainer by removing hardcoded extension

import torch
W = torch.rand(d,n_neurons)
b = torch.rand(n_neurons,1)
X = torch.rand(n,d)
with ts.explain(savefig="my_inspection.pdf"):
    Y = W @ X.T + b

[parrt]

Hi. I can't remember why but something prevented me from allowing non-svg. Hmm...let me look.

[parrt]

ok, looks good. We only save with plt.savefig() in one spot, PyVizView.savefig(), so should be safe to allow any extensions. thanks!

[parrt]

Confirmed that this works with pdf now:

import torch
import tsensor

n = 100
d = 10
n_neurons = 100
W = torch.rand(d,n_neurons)
b = torch.rand(n_neurons,1)
X = torch.rand(n,d)
with tsensor.explain(savefig="my_inspection.pdf"):
    Y = W @ X.T + b