cleanup

Author: CarloLucibello

src/rulesets/SparseArrays/sparsematrix.jl

@@ -29,14 +29,8 @@ function rrule(::typeof(findnz), A::AbstractSparseMatrix)
     m, n = size(A)
 
     function findnz_pullback(Δ)
-        Δ === NoTangent() && return (NoTangent(), Δ)
-        Δ === ZeroTangent() && return (NoTangent(), Δ)
-        
         _, _, V̄ = unthunk(Δ)
-        
-        V̄ === NoTangent() && return (NoTangent(), V̄)
-        V̄ === ZeroTangent() && return (NoTangent(), V̄)
-        
+        V̄ isa AbstractZero && return (NoTangent(), V̄)
         return NoTangent(), sparse(I, J, V̄, m, n)
     end
 
@@ -49,10 +43,7 @@ function rrule(::typeof(findnz), v::AbstractSparseVector)
 
     function findnz_pullback(Δ)
         _, V̄ = unthunk(Δ)
-
-        V̄ === NoTangent() && return (NoTangent(), V̄)
-        V̄ === ZeroTangent() && return (NoTangent(), V̄)
-        
+        V̄ isa AbstractZero && return (NoTangent(), V̄)
         return NoTangent(), sparse(I, V̄, n)
     end