Update pymc_extras/inference/deterministic_advi/dadvi.py

Co-authored-by: Jesse Grabowski <[email protected]>

Author: martiningram

pymc_extras/inference/deterministic_advi/dadvi.py

@@ -152,8 +152,7 @@ def create_dadvi_graph(
 
     var_params = pt.vector(name="eta", shape=(2 * n_params,))
 
-    means = var_params[:n_params]
-    log_sds = var_params[n_params:]
+    means , log_sds= pt.split(var_params, 2)
 
     draw_matrix = pt.constant(draws)
     samples = means + pt.exp(log_sds) * draw_matrix