Bug fix: Needed to add Flux to Project.toml

spm · Mar 14, 2024 · 1a78ec1 · 1a78ec1
1 parent fce0b8d
commit 1a78ec1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -8,9 +8,9 @@ Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 FFTW = "7a1cc6ca-52ef-59f5-83cd-3a7055c09341"
+Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
 LazyArtifacts = "4af54fe1-eca0-43a8-85a7-787d91b784e3"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
-Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [extras]

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,6 +1,8 @@
 using PushPull
 using Test
 using CUDA
+using Flux
+
 function gpu_cpu_K(d::NTuple{3,Int64}, reg::Vector{<:AbstractFloat})
     vx = [1.2, 0.7, 1.5]
     u  = randn(Float32,(d...,3))
@@ -50,7 +52,6 @@ function operator_consistency(d::NTuple{3,Int64}, reg::Vector{<:AbstractFloat},
     return sum((u1[:].-u0[:]).^2)./sum(u0[:].^2)
 end
 
-using Flux
 function test_grad(fun,ϕ)
 
     function numerical_grad(fun, θ₀)