Fixed radial also in AD-compatibility tests

SciML · May 13, 2020 · 0bce50c · 0bce50c
1 parent c0bd7f7
commit 0bce50c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/AD_compatibility.jl b/test/AD_compatibility.jl
@@ -193,7 +193,7 @@ f = x -> x^2
 y = f.(x)
 
 #Radials
-my_rad = RadialBasis(x,y,lb,ub,x->norm(x),2)
+my_rad = RadialBasis(x,y,lb,ub,linearRadial)
 g = x -> my_rad'(x)
 g(5.0)
 
@@ -245,7 +245,7 @@ f = x -> x[1]*x[2]
 y = f.(x)
 
 #Radials
-my_rad = RadialBasis(x,y,[lb,ub],z->norm(z),2)
+my_rad = RadialBasis(x,y,lb,ub,linearRadial)
 g = x -> Zygote.gradient(my_rad,x)
 g((2.0,5.0))
 
@@ -308,7 +308,7 @@ n_echos = 1
 my_neural = NeuralSurrogate(x,y,lb,ub,model,loss,opt,n_echos)
 Zygote.gradient(x -> sum(my_neural(x)), (2.0, 5.0))
 
-my_rad = RadialBasis(x,y,[lb,ub],z->norm(z),1)
+my_rad = RadialBasis(x,y,lb,ub,linearRadial)
 Zygote.gradient(x -> sum(my_rad(x)), (2.0, 5.0))
 
 p = 1.4