Merge branch 'master' into fixes_threaded

Author: devmotion

src/compiler.jl

@@ -352,7 +352,7 @@ function build_output(model_info)
         $generator($(args...)) = $(DynamicPPL.Model)($evaluator, $args_nt, $model_gen_constructor)
         $(generator_kw_form...)
 
-        $model_gen = $model_gen_constructor
+        $(Base).@__doc__ $model_gen = $model_gen_constructor
     end
 end
 

test/compiler.jl

@@ -580,4 +580,18 @@ end
         model = demo()
         @test all(iszero(model()) for _ in 1:1000)
     end
+    @testset "docstring" begin
+        "This is a test"
+        @model function demo(x)
+            m ~ Normal()
+            x ~ Normal(m, 1)
+        end
+
+        s = @doc(demo)
+        @test string(s) == "This is a test\n"
+
+        # Verify that adding docstring didn't completely break execution of model
+        m = demo(0.)
+        @test m() isa Float64
+    end
 end