Update SciMLSensitivity compats

LuxDL · May 14, 2024 · 5b21677 · 5b21677
1 parent 97b72b1
commit 5b21677
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 6 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Lux"
 uuid = "b2108857-7c20-44ae-9111-449ecde12c47"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "0.5.47"
+version = "0.5.48"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

diff --git a/docs/run_single_tutorial.jl b/docs/run_single_tutorial.jl
@@ -10,6 +10,8 @@ output_directory = ARGS[2]
 path = ARGS[3]
 
 io = open(pkg_log_path, "w")
+Pkg.Registry.update()
+Pkg.update()
 Pkg.develop(; path=joinpath(@__DIR__, ".."), io)
 Pkg.instantiate(; io)
 close(io)

diff --git a/examples/NeuralODE/Project.toml b/examples/NeuralODE/Project.toml
@@ -27,6 +27,6 @@ MLUtils = "0.2, 0.3, 0.4"
 OneHotArrays = "0.1, 0.2"
 Optimisers = "0.2, 0.3"
 OrdinaryDiffEq = "6"
-SciMLSensitivity = "7.45"
+SciMLSensitivity = "7"
 Statistics = "1"
 Zygote = "0.6"
diff --git a/examples/PolynomialFitting/main.jl b/examples/PolynomialFitting/main.jl
@@ -74,12 +74,12 @@ vjp_rule = AutoZygote()
 function main(tstate::Lux.Experimental.TrainState, vjp, data, epochs)
  data = data .|> gpu_device()
  for epoch in 1:epochs
- grads, loss, stats, tstate = Lux.Training.compute_gradients(
+ grads, loss, stats, tstate = Lux.Experimental.compute_gradients(
  vjp, loss_function, data, tstate)
  if epoch % 50 == 1 || epoch == epochs
  @printf "Epoch: %3d \t Loss: %.5g\n" epoch loss
  end
- tstate = Lux.Training.apply_gradients!(tstate, grads)
+ tstate = Lux.Experimental.apply_gradients!(tstate, grads)
  end
  return tstate
 end

diff --git a/examples/SymbolicOptimalControl/Project.toml b/examples/SymbolicOptimalControl/Project.toml
@@ -30,7 +30,7 @@ Optimization = "3.24.3"
 OptimizationOptimJL = "0.2.3"
 OptimizationOptimisers = "0.2.1"
 OrdinaryDiffEq = "6.74.1"
-SciMLSensitivity = "7.56.2"
-Statistics = "1.11.1"
+SciMLSensitivity = "7"
+Statistics = "1.11"
 SymbolicRegression = "0.24.1"
 SymbolicUtils = "1.5.1"