def METHOD_NAME(self) -> CompressionStage:
    """
    Returns the compression stage. Should be used on saving best checkpoints
    to distinguish between uncompressed, partially compressed, and fully
    compressed models.
    :return: The compression stage of the target model.
    """
    return CompressionStage.UNCOMPRESSED