Reduce the reach of type-piracy

bsc-quantic · Sep 10, 2023 · f335f50 · f335f50
1 parent fe93cde
commit f335f50
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ext/TenetChainRulesCoreExt.jl b/ext/TenetChainRulesCoreExt.jl
@@ -21,9 +21,9 @@ function ChainRulesCore.rrule(T::Type{<:Tensor}, data, inds; meta...)
 end
 
 # WARN type-piracy
-@non_differentiable setdiff(s, itrs...)
-@non_differentiable union(s, itrs...)
-@non_differentiable intersect(s, itrs...)
-@non_differentiable symdiff(s, itrs...)
+@non_differentiable setdiff(s::Base.AbstractVecOrTuple{Symbol}, itrs::Base.AbstractVecOrTuple{Symbol}...)
+@non_differentiable union(s::Base.AbstractVecOrTuple{Symbol}, itrs::Base.AbstractVecOrTuple{Symbol}...)
+@non_differentiable intersect(s::Base.AbstractVecOrTuple{Symbol}, itrs::Base.AbstractVecOrTuple{Symbol}...)
+@non_differentiable symdiff(s::Base.AbstractVecOrTuple{Symbol}, itrs::Base.AbstractVecOrTuple{Symbol}...)
 
 end