fix: remove global imports

LuxDL · Aug 24, 2024 · 9842802 · 9842802
1 parent d3d3b21
commit 9842802
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/Boltz.jl b/src/Boltz.jl
@@ -1,14 +1,9 @@
 module Boltz
 
 using ArgCheck: @argcheck
-using Artifacts: Artifacts, @artifact_str
-using JLD2: JLD2, load
-using ForwardDiff: ForwardDiff
-using GPUArraysCore: GPUArraysCore
-using LazyArtifacts: LazyArtifacts
+using JLD2: JLD2 # TODO[BREAKING]: Remove JLD2 dependency and ask users to load it
 using Random: Random
 using Reexport: @reexport
-using Statistics: mean
 
 @reexport using Lux
 

diff --git a/src/initialize.jl b/src/initialize.jl
@@ -2,7 +2,8 @@ module InitializeModels
 
 using ArgCheck: @argcheck
 using Artifacts: Artifacts, @artifact_str
-using JLD2: JLD2
+using JLD2: JLD2 # TODO[BREAKING]: Remove JLD2 dependency and ask users to load it
+using LazyArtifacts: LazyArtifacts
 using Random: Random
 
 using LuxCore: LuxCore