Skip to content

Commit

Permalink
fixed typo: ml.complete -> ml.compile {mlflow#13909} (mlflow#13922)
Browse files Browse the repository at this point in the history
Signed-off-by: S Ajay <[email protected]>
Co-authored-by: S Ajay <[email protected]>
  • Loading branch information
Ajay-Satish-01 and Ajay-Satish-01 authored Nov 29, 2024
1 parent 4800875 commit 64721e7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mlflow/tensorflow/callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,7 +207,7 @@ def on_epoch_begin(self, epoch, logs=None):

def on_train_batch_end(self, batch, logs=None):
# Note that `on_train_batch_end` might be invoked by every N train steps,
# (controlled by `steps_per_execution` argument in `model.complete` method).
# (controlled by `steps_per_execution` argument in `model.compile` method).
# the following logic is similar to
# https://github.com/keras-team/keras/blob/e6e62405fa1b4444102601636d871610d91e5783/keras/callbacks/model_checkpoint.py#L212
add_batches = batch + 1 if batch <= self._last_batch_seen else batch - self._last_batch_seen
Expand Down

0 comments on commit 64721e7

Please sign in to comment.