From 538bafe1722ac4852541bc9786c1fc7f275c6460 Mon Sep 17 00:00:00 2001 From: Ray Kim Date: Wed, 12 Jun 2024 23:53:50 +0100 Subject: [PATCH] fix signature for projection of `LocationScale` families --- src/families/location_scale.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/families/location_scale.jl b/src/families/location_scale.jl index 5c3b0eda..1e96576f 100644 --- a/src/families/location_scale.jl +++ b/src/families/location_scale.jl @@ -168,7 +168,7 @@ function MeanFieldGaussian( end function update_variational_params!( - ::MvLocationScale, opt_st, params, restructure, grad + ::Type{<:MvLocationScale}, opt_st, params, restructure, grad ) opt_st, params = Optimisers.update!(opt_st, params, grad) q = restructure(params)