From b110c1a7a5a27b6f6f5fbf8556e50c4c016be439 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Fri, 8 Mar 2019 13:12:06 +0000
Subject: [PATCH] zygote fixup

---
 src/Flux.jl | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/Flux.jl b/src/Flux.jl
index fd88b41b55..e3c3ad89e3 100644
--- a/src/Flux.jl
+++ b/src/Flux.jl
@@ -5,15 +5,13 @@ module Flux
 using Base: tail
 using MacroTools, Juno, Requires, Reexport, Statistics, Random
 using MacroTools: @forward
+@reexport using NNlib
+using Zygote: gradient, @adjoint
 
 export Chain, Dense, RNN, LSTM, GRU, Conv, ConvTranspose, MaxPool, MeanPool,
        DepthwiseConv, Dropout, AlphaDropout, LayerNorm, BatchNorm, InstanceNorm,
        params, mapleaves, cpu, gpu, f32, f64
 
-@reexport using NNlib
-
-using Zygote
-
 include("optimise/Optimise.jl")
 using .Optimise
 using .Optimise: @epochs