From 4b9a2b865d68f5ce58542db439f3d2ca7eff34d3 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Mon, 22 Aug 2016 14:49:34 +0100
Subject: [PATCH] model macro

---
 src/compiler/code.jl | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/compiler/code.jl b/src/compiler/code.jl
index 7ed226b9..1b76a194 100644
--- a/src/compiler/code.jl
+++ b/src/compiler/code.jl
@@ -71,12 +71,12 @@ function process_type(ex)
     (self::$T)($(args...),) = $(syntax(build_forward(body, args)))
     back!(self::$T, Δ, $(args...)) = $(syntax(build_backward(body, args[1], params)))
     $(build_update(T, params))
-  end |> longdef
     graph(::$T) = $(Flow.constructor(makegraph(body, args)))
+    nothing
+  end |> esc
 end
 
-# process_type(:(type Sigmoid
-#   W
-#   b
-#   x -> σ(W*x+b)
-# end)) |> prettify
+macro model(ex)
+  isexpr(ex, :type) ? process_type(ex) :
+  error("Unsupported model expression $ex")
+end