➕ Include in top-level only

FluxML · Sep 1, 2024 · 59c9bb9 · 59c9bb9
1 parent daa4b2a
commit 59c9bb9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/MLJFlux.jl b/src/MLJFlux.jl
@@ -27,6 +27,8 @@ include("core.jl")
 include("regressor.jl")
 include("classifier.jl")
 include("image.jl")
+include("fit_utils.jl")
+include("entity_embedding_utils.jl")
 include("mlj_model_interface.jl")
 
 export NeuralNetworkRegressor, MultitargetNeuralNetworkRegressor

diff --git a/src/mlj_model_interface.jl b/src/mlj_model_interface.jl
@@ -55,8 +55,6 @@ end
 
 
 # # FIT AND  UPDATE
-include("fit_utils.jl")
-include("entity_embedding_utils.jl")
 
 const ERR_BUILDER = 
     "Builder does not appear to build an architecture compatible with supplied data. "