👨‍🔧 Fix small error

FluxML · Aug 5, 2024 · af91ee9 · af91ee9
1 parent d1e65c0
commit af91ee9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/MLJFlux.jl b/src/MLJFlux.jl
@@ -1,7 +1,7 @@
 module MLJFlux
 
 export CUDALibs, CPU1
-
+import Flux
 using MLJModelInterface
 using MLJModelInterface.ScientificTypesBase
 using ScientificTypes: schema, Finite
@@ -13,6 +13,8 @@ using Statistics
 using ColorTypes
 using ComputationalResources
 using Random
+import Metalhead
+import Optimisers
 
 include("utilities.jl")
 const MMI = MLJModelInterface