lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add programmatic docstring updates

Browse Source
This commit is contained in:
Francois Chollet 2023-06-12 14:05:19 -07:00
parent 2114c02b9c
commit fba2f770a4
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
keras_core/models/model.py
View File

@ -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__

1
keras_core/optimizers/optimizer.py
View File

@ -18,3 +18,4 @@ class Optimizer(BackendOptimizer):
base_optimizer_keyword_args = base_optimizer.base_optimizer_keyword_args
Optimizer.__doc__ = base_optimizer.BaseOptimizer.__doc__