Tweak API for update (#17)

* Tweak API for update

* New dev version

* remove -DEV

* Allow tests to fail on nightly

* Fix dep

Author: willtebbutt

.travis.yml

@@ -6,6 +6,9 @@ julia:
   - 1.0
   - 1.3
   - nightly
+jobs:
+  allow_failures:
+    - julia: nightly
 
 notifications:
   email:
@@ -16,4 +19,4 @@ notifications:
     if: type = cron
 matrix:
   allow_failures:
-    - julia: nightly
+    - julia: nightly

Project.toml

@@ -1,6 +1,6 @@
 name = "ChainRulesTestUtils"
 uuid = "cdddcdb0-9152-4a09-a978-84456f9df70a"
-version = "0.1.3"
+version = "0.2.0"
 
 [deps]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
@@ -9,7 +9,7 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [compat]
-ChainRulesCore = "0.5, 0.6"
+ChainRulesCore = "0.7"
 FiniteDifferences = "0.7, 0.8, 0.9"
 julia = "1"
 

src/ChainRulesTestUtils.jl

@@ -63,7 +63,7 @@ function test_scalar(f, x; rtol=1e-9, atol=1e-9, fdm=_fdm, kwargs...)
     ensure_not_running_on_functor(f, "test_scalar")
 
     r_res = rrule(f, x)
-    f_res = frule(f, x, Zero(), 1)
+    f_res = frule((Zero(), 1), f, x)
     @test r_res !== nothing  # Check the rule was defined
     @test f_res !== nothing
     r_fx, prop_rule = r_res
@@ -111,7 +111,7 @@ end
 function frule_test(f, xẋs::Tuple{Any, Any}...; rtol=1e-9, atol=1e-9, fdm=_fdm, kwargs...)
     ensure_not_running_on_functor(f, "frule_test")
     xs, ẋs = collect(zip(xẋs...))
-    Ω, dΩ_ad = frule(f, xs..., NO_FIELDS, ẋs...)
+    Ω, dΩ_ad = frule((NO_FIELDS, ẋs...), f, xs...)
     @test f(xs...) == Ω
 
     # Correctness testing via finite differencing.

test/runtests.jl

@@ -9,7 +9,7 @@ using Test
     test_scalar(double, 2)
 
     fst(x, y) = x
-    ChainRulesCore.frule(::typeof(fst), x, y, _, dx, dy) = (x, dx)
+    ChainRulesCore.frule((_, dx, dy), ::typeof(fst), x, y) = (x, dx)
 
     function ChainRulesCore.rrule(::typeof(fst), x, y)
         function fst_pullback(Δx)