josipd / torch-two-sample

A PyTorch library for two-sample tests
Other
237 stars 33 forks source link

Deprecated usage of torch.gesv() needs to be fixed #12

Open nkamath5 opened 1 year ago

nkamath5 commented 1 year ago

Thanks for this amazing library! One quick update required for the code though:

In inferencetrees.py, line 121, ```Z, = torch.gesv(A, L * P.expand_as(L))`` is used to solve a system of equations. However torch.gesv() is deprecated versions 1.1.0 & above, at https://pytorch.org/docs/1.1.0/torch.html#torch.gesv they suggest usingtorch.solve()` instead.