enum support (#293)

* enum support * add test * bump version
compintell · Oct 20, 2024 · faa6397 · faa6397 · willtebbutt · Oct 20, 2024
1 parent 50e8d9c
commit faa6397
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Mooncake"
 uuid = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
 authors = ["Will Tebbutt, Hong Ge, and contributors"]
-version = "0.4.13"
+version = "0.4.14"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

diff --git a/src/tangents.jl b/src/tangents.jl
@@ -332,8 +332,9 @@ tangent_type(::Type{DimensionMismatch}) = NoTangent
 
 tangent_type(::Type{Method}) = NoTangent
 
-function tangent_type(::Type{P}) where {N, P<:Tuple{Vararg{Any, N}}}
+tangent_type(::Type{<:Enum}) = NoTangent
 
+function tangent_type(::Type{P}) where {N, P<:Tuple{Vararg{Any, N}}}
     # As with other types, tangent type of Union is Union of tangent types.
     P isa Union && return Union{tangent_type(P.a), tangent_type(P.b)}
 
@@ -374,7 +375,6 @@ function tangent_type(::Type{P}) where {N, P<:NamedTuple{N}}
 end
 
 @generated function tangent_type(::Type{P}) where {P}
-
     # This method can only handle struct types. Tell user to implement tangent type
     # directly for primitive types.
     isprimitivetype(P) && throw(error(
@@ -950,6 +950,7 @@ function tangent_test_cases()
         (a=randn(10), b=3),
         (a=3, b=randn(10)),
         (a=randn(10), b=randn(10)),
+        (Base.TOML.ErrorType(1), NoTangent()), # Enum
     ]
     return vcat(
         map(x -> (false, x...), abs_test_cases),