From fba2f770a4cb15e4aa9ca31a0116ec5d3baa497c Mon Sep 17 00:00:00 2001
From: Francois Chollet <francois.chollet@gmail.com>
Date: Mon, 12 Jun 2023 14:05:19 -0700
Subject: [PATCH] Add programmatic docstring updates

---
 keras_core/models/model.py         | 9 +++++++++
 keras_core/optimizers/optimizer.py | 1 +
 2 files changed, 10 insertions(+)

diff --git a/keras_core/models/model.py b/keras_core/models/model.py
index 76845df39..6bc32aeae 100644
--- a/keras_core/models/model.py
+++ b/keras_core/models/model.py
@@ -8,6 +8,7 @@ from keras_core.api_export import keras_core_export
 from keras_core.layers.layer import Layer
 from keras_core.saving import saving_api
 from keras_core.saving import saving_lib
+from keras_core.trainers import trainer as base_trainer
 from keras_core.utils import io_utils
 from keras_core.utils import summary_utils
 from keras_core.utils import traceback_utils
@@ -436,3 +437,11 @@ def inject_functional_model_class(cls):
     cls.__new__(cls)
 
     return cls
+
+
+Model.fit.__doc__ = base_trainer.Trainer.fit.__doc__
+Model.predict.__doc__ = base_trainer.Trainer.predict.__doc__
+Model.evaluate.__doc__ = base_trainer.Trainer.evaluate.__doc__
+Model.train_on_batch.__doc__ = base_trainer.Trainer.train_on_batch.__doc__
+Model.test_on_batch.__doc__ = base_trainer.Trainer.test_on_batch.__doc__
+Model.predict_on_batch.__doc__ = base_trainer.Trainer.predict_on_batch.__doc__
diff --git a/keras_core/optimizers/optimizer.py b/keras_core/optimizers/optimizer.py
index dc9c596d6..ca81b34fb 100644
--- a/keras_core/optimizers/optimizer.py
+++ b/keras_core/optimizers/optimizer.py
@@ -18,3 +18,4 @@ class Optimizer(BackendOptimizer):
 
 
 base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args
+Optimizer.__doc__ = base_optimizer.BaseOptimizer.__doc__