fix signature for projection of LocationScale families

Red-Portal · Red-Portal · commit 538bafe1722a · 2024-06-12T23:53:50.000+01:00
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)

Original file line number	Diff line number	Diff line change
`@@ -168,7 +168,7 @@ function MeanFieldGaussian(`
`168`	`168`	`end`
`169`	`169`
`170`	`170`	`function update_variational_params!(`
`171`		`- ::MvLocationScale, opt_st, params, restructure, grad`
	`171`	`+ ::Type{<:MvLocationScale}, opt_st, params, restructure, grad`
`172`	`172`	`)`
`173`	`173`	`opt_st, params = Optimisers.update!(opt_st, params, grad)`
`174`	`174`	`q = restructure(params)`