Update DifferentiationInterface dependency to v0.6 (#181)

Author: adrhill

Project.toml

@@ -14,7 +14,7 @@ XAIBase = "9b48221d-a747-4c1b-9860-46a1d8ba24a7"
 
 [compat]
 ADTypes = "1"
-DifferentiationInterface = "0.5"
+DifferentiationInterface = "0.6"
 Distributions = "0.25"
 Random = "<0.0.1, 1"
 Reexport = "1"

src/gradient.jl

@@ -18,7 +18,8 @@ function gradient_wrt_input(
     dy = zero(output)
     dy[output_selection] .= 1
 
-    output, grad = value_and_pullback(model, backend, input, dy)
+    output, pbs = value_and_pullback(model, backend, input, tuple(dy))
+    grad = only(pbs)
     return grad, output, output_selection
 end