chore: update docs project compat

LuxDL · Aug 27, 2024 · 79d4772 · 79d4772
1 parent c18fdbf
commit 79d4772
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/docs/Project.toml b/docs/Project.toml
@@ -10,7 +10,7 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [compat]
-Boltz = "0.3"
+Boltz = "0.4"
 Documenter = "1"
 DocumenterCitations = "1"
 DocumenterVitepress = "0.1"

diff --git a/test/layer_tests.jl b/test/layer_tests.jl
@@ -28,7 +28,7 @@
 end
 
 @testitem "Hamiltonian Neural Network" setup=[SharedTestSetup] tags=[:layers] begin
-    using ComponentArrays, ForwardDiff, Zygote
+    using ComponentArrays, ForwardDiff, Zygote, MLDataDevices
 
     _remove_nothing(xs) = map(x -> x === nothing ? 0 : x, xs)
 
@@ -150,7 +150,7 @@ end
 end
 
 @testitem "Spline Layer" setup=[SharedTestSetup] tags=[:layers] begin
-    using ComponentArrays, DataInterpolations, ForwardDiff, Zygote
+    using ComponentArrays, DataInterpolations, ForwardDiff, Zygote, MLDataDevices
 
     @testset "$(mode)" for (mode, aType, dev, ongpu) in MODES
         ongpu && continue

diff --git a/test/shared_testsetup.jl b/test/shared_testsetup.jl
@@ -4,8 +4,8 @@ using Enzyme
 Enzyme.API.runtimeActivity!(true)
 
 import Reexport: @reexport
-@reexport using Boltz, Lux, GPUArraysCore, LuxLib, LuxTestUtils, Random, StableRNGs,
-                MLDataDevices
+@reexport using Boltz, Lux, GPUArraysCore, LuxLib, LuxTestUtils, Random, StableRNGs
+using MLDataDevices
 import Metalhead
 
 LuxTestUtils.jet_target_modules!(["Boltz", "Lux", "LuxLib"])

diff --git a/test/vision_tests.jl b/test/vision_tests.jl
@@ -114,8 +114,7 @@ end
         @jet model(img, ps, st)
         @test size(first(model(img, ps, st))) == (1000, 2)
 
-        # @test_deprecated doesnt work since other @warn s are present
-        model, ps, st = vision_transformer(name; pretrained=false)
+        model, ps, st = Vision.VisionTransformer(name; pretrained=false)
         ps = ps |> dev
         st = Lux.testmode(st) |> dev
         img = randn(Float32, 256, 256, 3, 2) |> aType