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","."]
code: def on_visualize_figure(self, fig):
      logger.info("aim.on_visualize_figure() called...")
      if self.aim_run:
          self.aim_run.track(
              aim.Figure(fig),
              name="Figure",
              context={"type": "Training Figure"},
          )
  
file_path: ludwig/ludwig/contribs/aim.py
imports: import json
  import logging
  from ludwig.api_annotations import PublicAPI
  from ludwig.callbacks import Callback
  from ludwig.utils.data_utils import NumpyEncoder
  from ludwig.utils.package_utils import LazyLoader
label_desc: Use image node_id 9 for calling the AimCallback obj's underlying member method code with example usage: obj.on_visualize_figure(fig) without return types