Fix package name printing (#517)

JuliaDiff · Sep 30, 2024 · 3f29b61 · 3f29b61
1 parent 7a09754
commit 3f29b61
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/DifferentiationInterface/src/utils/printing.jl b/DifferentiationInterface/src/utils/printing.jl
@@ -1,8 +1,10 @@
 function package_name(b::AbstractADType)
     s = string(b)
+    s = chopprefix(s, "ADTypes.")
+    s = chopprefix(s, "Auto")
     k = findfirst('(', s)
     isnothing(k) && throw(ArgumentError("Cannot parse backend into package"))
-    return s[5:(k - 1)]
+    return s[begin:(k - 1)]
 end
 
 function package_name(b::SecondOrder)