Merge pull request #69 from hmyuuu/fix-deps

Fix deps
QuanEstimation · Sep 1, 2022 · 4716f27 · 4716f27
2 parents d9ddf61 + f905a27
commit 4716f27
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -26,11 +26,11 @@ Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
 BoundaryValueDiffEq = "2"
-Convex = "0.14, 0.15"
+Convex = "0.15"
 Distributions = "0.25"
-Flux = "0.12, 0.14"
-Interpolations = "0.13, 0.15"
-IntervalSets = "0.5, 0.7"
+Flux = "0.12, 0.13"
+Interpolations = "0.13"
+IntervalSets = "0.5"
 QuadGK = "2"
 ReinforcementLearning = "0.10"
 SCS = "1"

diff --git a/src/Algorithm/DDPG.jl b/src/Algorithm/DDPG.jl
@@ -1,9 +1,11 @@
 ## TODO: Need reconstruct !!!
 using Flux: glorot_normal, glorot_uniform
 using StableRNGs
-using Flux.Losses
 using IntervalSets
 
+Base.copyto!(dest::ReinforcementLearning.NeuralNetworkApproximator, src::ReinforcementLearning.NeuralNetworkApproximator) =
+           Flux.loadparams!(dest.model, Flux.params(src))
+
 function Base.rsplit( v, l::Int)
     u = reshape(v,l,length(v)÷l)
     [u[:,i] for i=1:size(u,2)]