fix signature for projection of LocationScale families

TuringLang · Jun 12, 2024 · 538bafe · 538bafe
1 parent a54c7fc
commit 538bafe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)