Fix closures

Author: oxinabox

src/stage1/forward.jl

@@ -107,6 +107,7 @@ function shuffle_base(r)
 end
 
 function (::∂☆internal{1})(args::AbstractTangentBundle{1}...)
+    # @info "∂☆internal{1}" args
     r = frule(#=DiffractorRuleConfig(),=# map(first_partial, args), map(primal, args)...)
     if r === nothing
         return ∂☆recurse{1}()(args...)
@@ -132,12 +133,12 @@ end
 function (::∂☆internal{N})(f::AbstractZeroBundle{N}, args::AbstractZeroBundle{N}...) where {N}
     f_v = primal(f)
     args_v = map(primal, args)
-    return ZeroBundle{N}(f_v(args_v...))
+    return zero_bundle{N}()(f_v(args_v...))
 end
 function (::∂☆internal{1})(f::AbstractZeroBundle{1}, args::AbstractZeroBundle{1}...)
     f_v = primal(f)
     args_v = map(primal, args)
-    return ZeroBundle{1}(f_v(args_v...))
+    return zero_bundle{1}()(f_v(args_v...))
 end
 
 function (::∂☆internal{N})(args::AbstractTangentBundle{N}...) where {N}

test/forward_mutation.jl

@@ -66,5 +66,22 @@ end
     @test 🐇[TaylorTangentIndex(2)].x == 2.0  # returns 20
 end
 
+@testset "closure" begin
+    function bar(x)
+        z = x + 1.0
+        function foo!(y)
+            z = z * y
+            return z
+        end
+
+        foo!(2)
+        foo!(2)
+        return z 
+    end
+
+    🥯 = ∂☆{1}()(ZeroBundle{1}(bar), TaylorBundle{1}(10.0, (1.0,)))
+    @test 🥯[TaylorTangentIndex(1)] == 4.0
+end
+
 
 # end # module