Fix BNN doc to work with Mooncake

Author: penelopeysm

tutorials/03-bayesian-neural-network/index.qmd

@@ -189,7 +189,7 @@ const nn = StatefulLuxLayer{true}(nn_initial, nothing, st)
     parameters ~ MvNormal(zeros(nparameters), Diagonal(abs2.(sigma .* ones(nparameters))))
 
     # Forward NN to make predictions
-    preds = Lux.apply(nn, xs, vector_to_parameters(parameters, ps))
+    preds = Lux.apply(nn, xs, f32(vector_to_parameters(parameters, ps)))
 
     # Observe each prediction.
     for i in eachindex(ts)