Make NoTangent have an eltype of itself.

JuliaDiff · Sep 17, 2024 · c52ac88 · c52ac88
1 parent 5369090
commit c52ac88
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/tangent_types/abstract_zero.jl b/src/tangent_types/abstract_zero.jl
@@ -93,6 +93,8 @@ end
 """
 struct NoTangent <: AbstractZero end
 
+Base.eltype(::Type{NoTangent}) = NoTangent
+
 """
     zero_tangent(primal)
 

diff --git a/test/tangent_types/abstract_zero.jl b/test/tangent_types/abstract_zero.jl
@@ -119,6 +119,9 @@
         @test dot(dne, 17.2) == dne
         @test dot(11.9, dne) == dne
 
+        @test eltype(dne) === NoTangent
+        @test eltype(NoTangent) === NoTangent
+
         @test promote_type(NoTangent, Bool) == Bool
         @test promote_type(Bool, NoTangent) == Bool
         @test promote_type(NoTangent, Int64) == Int64
-Original file line number
+Diff line change
@@ Expand Up / @@ -93,6 +93,8 @@ end @@
     """
     struct NoTangent <: AbstractZero end
+    Base.eltype(::Type{NoTangent}) = NoTangent
     """
         zero_tangent(primal)
@@ Expand Down @@