src/hyperpod_nemo_adapter/utils/callbacks/local_ckpt_io.py [118:127]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def save_checkpoint(
        self,
        checkpoint: Dict[str, Any],
        path: _PATH,
        storage_options: Optional[Any] = None,
        *a,
        **kw,
    ) -> None:
        trainer = storage_options
        assert isinstance(trainer, pl.Trainer)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/hyperpod_nemo_adapter/utils/callbacks/sharded_ckpt_io.py [36:45]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def save_checkpoint(
        self,
        checkpoint: Dict[str, Any],
        path: _PATH,
        storage_options: Optional[Any] = None,
        *a,
        **kw,
    ) -> None:
        trainer = storage_options
        assert isinstance(trainer, pl.Trainer)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -