class_docstr: Class that defines the methods necessary to hook into process.
class_docstr_tok: ["Class","that","defines","the","methods","necessary","to","hook","into","process","."]
docstr: Called from ludwig/models/model.py.
docstr_tok: ["Called","from","ludwig\/models\/model.py","."]
code: def on_epoch_end(self, trainer, progress_tracker, save_path):
      """Called from ludwig/models/model.py."""
      for key, value in progress_tracker.log_metrics().items():
          wandb.log({key: value})
  
file_path: ludwig/ludwig/contribs/wandb.py
imports: import logging
  import os
  from ludwig.api_annotations import PublicAPI
  from ludwig.callbacks import Callback
  from ludwig.utils.package_utils import LazyLoader
label_desc: Use image node_id 4 for calling the WandbCallback obj's underlying member method code with example usage: obj.on_epoch_end(trainer, progress_tracker, save_path) without return types