Skip to content

Commit 095a58b

Browse files
committed
update neg_binomial_lpmf for vectors and row vectors
1 parent db837e8 commit 095a58b

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

stan/math/prim/prob/neg_binomial_2_lpmf.hpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,9 +52,9 @@ inline return_type_t<T_location, T_precision> neg_binomial_2_lpmf(
5252
T_partials_return logp(0.0);
5353
auto ops_partials = make_partials_propagator(mu_ref, phi_ref);
5454

55-
auto n_vec = as_array_or_scalar(n_ref);
56-
auto mu_vec = as_array_or_scalar(mu_ref);
57-
auto phi_vec = as_array_or_scalar(phi_ref);
55+
auto n_vec = as_array_or_scalar(as_column_vector_or_scalar(n_ref));
56+
auto mu_vec = as_array_or_scalar(as_column_vector_or_scalar(mu_ref));
57+
auto phi_vec = as_array_or_scalar(as_column_vector_or_scalar(phi_ref));
5858
decltype(auto) mu_val = value_of(mu_vec);
5959
decltype(auto) phi_val = value_of(phi_vec);
6060
auto log_phi = log(phi_val);

0 commit comments

Comments
 (0)