Merge pull request #1257 from stan-dev/feature/info-print-reject

Add print and reject to info functions output

Author: WardBrian

src/frontend/Info.ml

@@ -53,6 +53,8 @@ let rec get_function_calls_stmt ud_dists (funs, distrs) stmt =
   let acc =
     match stmt.stmt with
     | NRFunApp (StanLib _, f, _) -> (Set.add funs f.name, distrs)
+    | Print _ -> (Set.add funs "print", distrs)
+    | Reject _ -> (Set.add funs "reject", distrs)
     | Tilde {distribution; _} ->
         let possible_names =
           List.map ~f:(( ^ ) distribution.name) Utils.distribution_suffices

test/integration/cli-args/info/info.expected

@@ -26,7 +26,7 @@
   },
   "transformed parameters": { "t": { "type": "real", "dimensions": 2 } },
   "generated quantities": { "u": { "type": "real", "dimensions": 0 } },
-  "functions": [ "log", "reduce_sum", "sin", "square" ],
+  "functions": [ "log", "print", "reduce_sum", "reject", "sin", "square" ],
   "distributions": [
     "dirichlet_lupdf", "normal_lccdf", "normal_lcdf", "normal_lpdf",
     "std_normal_lupdf"

test/integration/cli-args/info/info.stan

@@ -57,6 +57,10 @@ model {
     target += std_normal_lupdf(y);
     target += reduce_sum(f_lpdf, g, 1);
     y ~ goo();
+    print("hello world");
+    if (0) {
+      reject("goodbye");
+    }
 }
 
 generated quantities {