class_docstr: Base class for Air's Keras callbacks.
code: def on_predict_batch_end(self, batch, logs=None):
      if "predict_batch_end" in self._on:
          self._handle(logs, "predict_batch_end")
  
imports: import shutil
  from typing import Dict, List, Optional, Union
  from tensorflow.keras.callbacks import Callback
  import ray
  from ray.train.tensorflow import TensorflowCheckpoint
  from ray.util.annotations import PublicAPI
label_desc: Use image node_id 10 for calling the _Callback obj's underlying member method code with example usage: obj.on_predict_batch_end(batch, logs) without return types