Refactor: Don't need all of Flux.jl - only Zygote.jl

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

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -8,10 +8,10 @@ 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"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [extras]
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,7 +1,7 @@
 using PushPull
 using Test
 using CUDA
-using Flux
+using Zygote
 
 function gpu_cpu_K(d::NTuple{3,Int64}, reg::Vector{<:AbstractFloat})
     vx = [1.2, 0.7, 1.5]