From 1bd58ddd9579e80f8402347241943964437cd6b0 Mon Sep 17 00:00:00 2001
From: pat-alt <altmeyerpat@gmail.com>
Date: Mon, 10 Jun 2024 09:09:32 +0200
Subject: [PATCH] deprecating fit! and train! methods

---
 src/MLJFlux.jl    |  1 +
 src/deprecated.jl | 19 +++++++++++++++++++
 2 files changed, 20 insertions(+)
 create mode 100644 src/deprecated.jl

diff --git a/src/MLJFlux.jl b/src/MLJFlux.jl
index 86ee5c28..1445ee78 100644
--- a/src/MLJFlux.jl
+++ b/src/MLJFlux.jl
@@ -32,6 +32,7 @@ export NeuralNetworkRegressor, MultitargetNeuralNetworkRegressor
 export NeuralNetworkClassifier, ImageClassifier
 export CUDALibs, CPU1
 
+include("deprecated.jl")
 
 
 end #module
diff --git a/src/deprecated.jl b/src/deprecated.jl
new file mode 100644
index 00000000..9b09cabd
--- /dev/null
+++ b/src/deprecated.jl
@@ -0,0 +1,19 @@
+Base.@deprecate fit!(model::MLJFlux.MLJFluxModel, penalty, chain, optimiser, epochs, verbosity, X, y) train(
+    model::MLJFlux.MLJFluxModel,
+    chain,
+    optimiser,
+    optimiser_state,
+    epochs,
+    verbosity,
+    X,
+    y,
+) false
+
+Base.@deprecate train!(model::MLJFlux.MLJFluxModel, penalty, chain, optimiser, X, y) train_epoch(
+    model::MLJFlux.MLJFluxModel,
+    chain,
+    optimiser,
+    optimiser_state,
+    X,
+    y,
+) false
\ No newline at end of file