From d91677f651a79bcb04e9c2f31e681ae9e6f85e07 Mon Sep 17 00:00:00 2001
From: Filippo Vicentini <filippovicentini@gmail.com>
Date: Sun, 29 Sep 2019 12:23:41 +0200
Subject: [PATCH] Fix `params!` to work with complex numbers

---
 src/functor.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/functor.jl b/src/functor.jl
index 73483ab9..f69f4a65 100644
--- a/src/functor.jl
+++ b/src/functor.jl
@@ -40,6 +40,7 @@ end
 trainable(m) = functor(m)[1]
 
 params!(p::Params, x::AbstractArray{<:Real}, seen = IdSet()) = push!(p, x)
+params!(p::Params, x::AbstractArray{Complex{<:Real}}, seen = IdSet()) = push!(p, x)
 
 function params!(p::Params, x, seen = IdSet())
   x in seen && return