[lightning.pytorch.callbacks_factory]
hyperpod_kpi_callback = sagemaker_hyperpod_observability_sdk.adapters.pytorch_lightning_adapter:setup_hyperpod_kpi_callback

[pytorch_lightning.callbacks_factory]
hyperpod_kpi_callback = sagemaker_hyperpod_observability_sdk.adapters.pytorch_lightning_adapter:setup_hyperpod_kpi_callback
