From 0802b4d5cfc49a6e41c2bc4bf04d6b71172368d1 Mon Sep 17 00:00:00 2001 From: Mike Innes Date: Tue, 6 Mar 2018 16:55:42 +0000 Subject: [PATCH] closes #198 --- src/treelike.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/treelike.jl b/src/treelike.jl index cac90e4e..fbe9fcad 100644 --- a/src/treelike.jl +++ b/src/treelike.jl @@ -8,8 +8,8 @@ mapchildren(f, x::Tuple) = map(f, x) function treelike(T, fs = fieldnames(T)) @eval current_module() begin - children(x::$T) = ($([:(x.$f) for f in fs]...),) - mapchildren(f, x::$T) = $T(f.(children(x))...) + Flux.children(x::$T) = ($([:(x.$f) for f in fs]...),) + Flux.mapchildren(f, x::$T) = $T(f.($children(x))...) end end