Merge pull request #1034 from adamhaber/feature/expose-bernoulli-logit-glm-rng

Expose bernoulli_logit_glm_rng

Author: WardBrian

src/middle/Stan_math_signatures.ml

@@ -934,6 +934,16 @@ let () =
     , ReturnType UReal
     , [UArray UInt; URowVector; UVector; UVector]
     , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [UMatrix; UVector; UVector]
+    , AoS ) ;
+  add_unqualified
+    ( "bernoulli_logit_glm_rng"
+    , ReturnType (UArray UInt)
+    , [URowVector; UVector; UVector]
+    , AoS ) ;
   add_binary_vec_int_real "bessel_first_kind" AoS ;
   add_binary_vec_int_real "bessel_second_kind" AoS ;
   add_binary_vec "beta" AoS ;

test/integration/good/function-signatures/distributions/pretty.expected

@@ -27,6 +27,10 @@ generated quantities {
   array[3] int N;
   array[3] int latent_states;
   matrix[3, 3] latent_states_prob;
+  matrix[2, 3] d_x_m = [[1, 2, 3], [4, 5, 6]];
+  row_vector[3] d_x_rv = [1, 2, 3];
+  vector[3] d_beta_v = [1, 2, 3]';
+  vector[2] d_alpha_v = [0.5, 0.6]';
   n = bernoulli_logit_rng(0.0);
   N = bernoulli_logit_rng(alpha);
   N = bernoulli_logit_rng(nu);
@@ -1729,5 +1733,7 @@ generated quantities {
   N = discrete_range_rng(n, N);
   N = discrete_range_rng(N, n);
   N = discrete_range_rng(N, N);
+  N = bernoulli_logit_glm_rng(d_x_m, d_alpha_v, d_beta_v);
+  N = bernoulli_logit_glm_rng(d_x_rv, d_alpha_v, d_beta_v);
 }
 

test/integration/good/function-signatures/distributions/rngs.stan

@@ -26,6 +26,10 @@ generated quantities {
   int N[3];
   int latent_states[3];
   matrix[3, 3] latent_states_prob;
+  matrix[2, 3] d_x_m = [[1, 2, 3], [4, 5, 6]];
+  row_vector[3] d_x_rv = [1, 2, 3];
+  vector[3] d_beta_v = [1, 2, 3]';
+  vector[2] d_alpha_v = [0.5, 0.6]';
   n = bernoulli_logit_rng(0.0);
   N = bernoulli_logit_rng(alpha);
   N = bernoulli_logit_rng(nu);
@@ -1728,5 +1732,7 @@ generated quantities {
   N = discrete_range_rng(n, N);
   N = discrete_range_rng(N, n);
   N = discrete_range_rng(N, N);
+  N = bernoulli_logit_glm_rng(d_x_m, d_alpha_v, d_beta_v);
+  N = bernoulli_logit_glm_rng(d_x_rv, d_alpha_v, d_beta_v);
 }
 

test/integration/good/function-signatures/distributions/univariate/discrete/bernoulli/bernoulli_logit_glm.stan

@@ -11,7 +11,6 @@ transformed data {
   real transformed_data_real;
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha_v, d_beta_v);
-  transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y| d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y| d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a| d_x_rv, d_alpha, d_beta_v);

test/integration/good/function-signatures/distributions/univariate/discrete/bernoulli/pretty.expected

@@ -177,7 +177,6 @@ transformed data {
   real transformed_data_real;
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a | d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a | d_x_m, d_alpha_v, d_beta_v);
-  transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a | d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y | d_x_m, d_alpha, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y | d_x_m, d_alpha_v, d_beta_v);
   transformed_data_real = bernoulli_logit_glm_lpmf(d_y_a | d_x_rv, d_alpha, d_beta_v);

test/integration/signatures/stan_math_sigs.expected

@@ -463,6 +463,8 @@ bernoulli_logit_glm_lpmf(array[] int, row_vector, real, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, row_vector, vector, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, matrix, real, vector) => real
 bernoulli_logit_glm_lpmf(array[] int, matrix, vector, vector) => real
+bernoulli_logit_glm_rng(row_vector, vector, vector) => array[] int
+bernoulli_logit_glm_rng(matrix, vector, vector) => array[] int
 bernoulli_logit_log(int, real) => real
 bernoulli_logit_log(int, vector) => real
 bernoulli_logit_log(int, row_vector) => real