Different dimension for pre-defined kwargs and data about stanify HOT 3 CLOSED

generated quantities{
    array[n_t] vector[n_r] prey_obs;
    array[n_t] vector[n_r] predator_obs;

    real pred_birth_frac[n_r] = normal_rng(rep_vector(0.05, R), 0.005);
    real m_noise_scale = normal_rng(0.01, 0.001);
    real prey_birth_frac = normal_rng(0.8, 0.08);

    // Initial ODE values
    real prey__init = 30;
    real process_noise__init = 0;
    real predator__init = 4;
    vector[3] initial_outcome;  // Initial ODE state vector
    initial_outcome[1] = prey__init;
    initial_outcome[2] = process_noise__init;
    initial_outcome[3] = predator__init;
    array[n_t] vector[n_r] prey;
    array[n_t] vector[n_r] predator;
    array[n_t] vector[n_r] process_noise;
    for (r in 1:n_r){
        array[n_t] vector[n_q] integrated_result = ode_rk45(vensim_ode_func, initial_outcome, initial_time, times, prey_birth_frac, pred_birth_frac[r], time_step, process_noise_scale);
        prey[:, r] = integrated_result[:, 1];
        process_noise[:, r]  = integrated_result[:, 2];
        predator[:, r]  = integrated_result[:, 3];
        prey_obs[:, r] = normal_rng(prey[:, r], m_noise_scale);
        predator_obs[:, r] = normal_rng(predator[:, r], m_noise_scale);
    }
}

draws2data_xr <xarray.Dataset>
Dimensions:              (draw: 5, chain: 1, initial_outcome_dim_0: 3,
                          integrated_result_dim_0: 200,
                          integrated_result_dim_1: 3, predator_dim_0: 200,
                          process_noise_dim_0: 200, prey_dim_0: 200,
                          prey_obs_dim_0: 200, predator_obs_dim_0: 200)
Coordinates:
  * chain                (chain) int64 1
  * draw                 (draw) int64 0 1 2 3 4
Dimensions without coordinates: initial_outcome_dim_0, integrated_result_dim_0,
                                integrated_result_dim_1, predator_dim_0,
                                process_noise_dim_0, prey_dim_0,
                                prey_obs_dim_0, predator_obs_dim_0
Data variables: (12/13)
    prey_birth_frac      (chain, draw) float64 0.8001 0.8738 0.6499 0.7694 0.734
    m_noise_scale        (chain, draw) float64 0.009252 0.0111 ... 0.01153
    pred_birth_frac      (chain, draw) float64 0.04873 0.05066 ... 0.05469
    predator__init       (chain, draw) float64 4.0 4.0 4.0 4.0 4.0
    process_noise__init  (chain, draw) float64 0.0 0.0 0.0 0.0 0.0
    prey__init           (chain, draw) float64 30.0 30.0 30.0 30.0 30.0
    ...                   ...
    integrated_result    (chain, draw, integrated_result_dim_0, integrated_result_dim_1) float64 ...
    predator             (chain, draw, predator_dim_0) float64 4.027 ... 5.747
    process_noise        (chain, draw, process_noise_dim_0) float64 0.004472 ...
    prey                 (chain, draw, prey_dim_0) float64 30.18 30.73 ... 3.724
    prey_obs             (chain, draw, prey_obs_dim_0) float64 30.19 ... 3.721
    predator_obs         (chain, draw, predator_obs_dim_0) float64 4.023 ... ...

 idata = az.from_cmdstanpy(
        prior=draws2data_data
    )
    print("let's see whether arviz assigns R correctly to prey_obs, prey etc", idata)
    print(idata.prior.prey_obs.shape)
    idata_orig = az.from_cmdstanpy(
        **idata_kwargs
    ).stack(prior_draw=["chain", "draw"], groups="prior_groups")
    idata_orig.reset_index("prior_draw", inplace=True)
    draws2data_xr = draws2data_data.draws_xr()
    print("draws2data_xr", draws2data_xr)
    print("dims says (1,5, 200) but where is n_r?", draws2data_xr.dims)
    idata_orig.add_groups(prior = draws2data_xr)

let's see whether arviz assigns R correctly to prey_obs, prey etc Inference data with groups:
	> prior
	> sample_stats_prior
Frozen({'chain': 1, 'draw': 5, 'initial_outcome_dim_0': 3, 'integrated_result_dim_0': 200, 'integrated_result_dim_1': 3, 'prey_dim_0': 200, 'process_noise_dim_0': 200, 'predator_dim_0': 200, 'prey_obs_dim_0': 200, 'predator_obs_dim_0': 200})

dims says (1,5, 200) but where is n_r? 
Frozen({'draw': 5, 'chain': 1, 'initial_outcome_dim_0': 3, 'integrated_result_dim_0': 200, 'integrated_result_dim_1': 3, 'prey_dim_0': 200, 'process_noise_dim_0': 200, 'predator_dim_0': 200, 'prey_obs_dim_0': 200, 'predator_obs_dim_0': 200})