JuliaGaussianProcesses / Stheno.jl

Probabilistic Programming with Gaussian processes in Julia
Other
338 stars 26 forks source link

Fix tests #243

Closed martincornejo closed 1 year ago

martincornejo commented 1 year ago

Partially fixes https://github.com/JuliaGaussianProcesses/Stheno.jl/issues/242

Related to changes made in https://github.com/JuliaGaussianProcesses/AbstractGPs.jl/pull/357

martincornejo commented 1 year ago

I've pushed a potential solution to solving AD with Zygote (since I don't know if a solution through ChainRules is viable). I have adapted the dispatch pattern of mean, cov and var to dispatch on all arguments instead of a tuple of the arguments. Maybe it's an ugly solution. Let me know what you think. (All tests are now passing on my machine)

martincornejo commented 1 year ago

Some auto-formatting changes were committed. Sorry for that.

willtebbutt commented 1 year ago

Closed by #244