add Returns for 1.6

FluxML · Sep 2, 2023 · dbf39d4 · dbf39d4
1 parent 419725c
commit dbf39d4
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/utils.jl b/src/utils.jl
@@ -143,3 +143,22 @@ end
 function rrule(cfg::RuleConfig{>:HasReverseMode}, ::typeof(_fast_broadcast!), f::F, x::AbstractArray, ys...)  where {F<:Function}
     rrule_via_ad(cfg, broadcast, f, x, ys...)
 end
+
+# Could get this from Compat.jl instead
+# https://github.com/JuliaLang/julia/pull/39794
+if VERSION < v"1.7.0-DEV.793"
+    struct Returns{V} <: Function
+        value::V
+        Returns{V}(value) where {V} = new{V}(value)
+        Returns(value) = new{Core.Typeof(value)}(value)
+    end
+
+    (obj::Returns)(args...; kw...) = obj.value
+    function Base.show(io::IO, obj::Returns)
+        show(io, typeof(obj))
+        print(io, "(")
+        show(io, obj.value)
+        print(io, ")")
+    end
+end
+