Fix stuff

Author: penelopeysm

ad.py

@@ -89,8 +89,12 @@ def run_ad(args):
         try:
             output = run_and_capture([*RUN_JULIA_COMMAND, "--run", model_key, adtype])
             result = try_float(output.splitlines()[-1])
+            if not isinstance(result, float):
+                print(f"Output: {output}")
         except sp.CalledProcessError as e:
-            print(f"Error running {model_key} with {adtype}. Output: {e.output}")
+            # Julia crashed
+            print(f"Julia crashed when running {model_key} with {adtype}.")
+            print(f"To reproduce, run: `julia --project=. main.jl --run {model_key} {adtype}`")
             result = "error"
 
         print(f" ... {model_key} with {adtype} ==> {result}")

main.jl

@@ -35,7 +35,7 @@ end
 # These imports tend to get used a lot in models
 using DynamicPPL: @model, to_submodel
 using Distributions
-using LinearAlgebra: I
+using LinearAlgebra
 
 include("models/assume_dirichlet.jl")
 include("models/assume_lkjcholu.jl")

models/demo_assume_dot_observe.jl

@@ -3,8 +3,6 @@
     s ~ InverseGamma(2, 3)
     m ~ Normal(0, sqrt(s))
     x .~ Normal(m, sqrt(s))
-
-    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_dot_observe()

models/demo_assume_dot_observe_literal.jl

@@ -3,8 +3,6 @@
     s ~ InverseGamma(2, 3)
     m ~ Normal(0, sqrt(s))
     [1.5, 2.0] .~ Normal(m, sqrt(s))
-
-    return (; s = s, m = m, x = [1.5, 2.0], logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_dot_observe_literal()

models/demo_assume_index_observe.jl

@@ -12,8 +12,6 @@
         m[i] ~ Normal(0, sqrt(s[i]))
     end
     x ~ MvNormal(m, Diagonal(s))
-
-    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_index_observe()

models/demo_assume_matrix_observe_matrix_index.jl

@@ -9,8 +9,6 @@
     m ~ MvNormal(zeros(n), Diagonal(s_vec))
 
     x[:, 1] ~ MvNormal(m, Diagonal(s_vec))
-
-    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_matrix_observe_matrix_index()

models/demo_assume_multivariate_observe.jl

@@ -3,8 +3,5 @@
     s ~ product_distribution([InverseGamma(2, 3), InverseGamma(2, 3)])
     m ~ MvNormal(zero(x), Diagonal(s))
     x ~ MvNormal(m, Diagonal(s))
-
-    return (; s = s, m = m, x = x, logp = getlogp(__varinfo__))
 end
-
 @register demo_assume_multivariate_observe()

models/demo_assume_multivariate_observe_literal.jl

@@ -3,8 +3,6 @@
     s ~ product_distribution([InverseGamma(2, 3), InverseGamma(2, 3)])
     m ~ MvNormal(zeros(2), Diagonal(s))
     [1.5, 2.0] ~ MvNormal(m, Diagonal(s))
-
-    return (; s = s, m = m, x = [1.5, 2.0], logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_multivariate_observe_literal()

models/demo_assume_observe_literal.jl

@@ -4,8 +4,6 @@
     m ~ Normal(0, sqrt(s))
     1.5 ~ Normal(m, sqrt(s))
     2.0 ~ Normal(m, sqrt(s))
-
-    return (; s = s, m = m, x = [1.5, 2.0], logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_observe_literal()

models/demo_assume_submodel_observe_index_literal.jl

@@ -12,8 +12,6 @@ end
     s, m = priors
     1.5 ~ Normal(m[1], sqrt(s[1]))
     2.0 ~ Normal(m[2], sqrt(s[2]))
-
-    return (; s = s, m = m, x = [1.5, 2.0], logp = getlogp(__varinfo__))
 end
 
 @register demo_assume_submodel_observe_index_literal()