Class CompressionConfig

examples/simple_trainer_old.py:113–144

@dataclass
class CompressionConfig:
    # Use PLAS sort in compression or not
    use_sort: bool = True
    # Verbose or not
    verbose: bool = True
    # QP value for video coding
    qp: Optional[int] = field(default=None)
    # Number of cluster of VQ for shN compression
    n_clusters: int = 32768
    # Maps attribute names to their codec functions
    attribute_codec_registry: Optional[AttributeCodecs] = field(default_factory=lambda: AttributeCodecs())

    def to_dict(self) -> Dict[str, Any]:
        """
        Convert the CompressionConfig instance to a dictionary.
        If attribute_codec_registry is not None, it will be converted to a dictionary using its to_dict method.
        Fields with None values (use_sort, verbose) will be excluded from the resulting dictionary.
        """
        result = {
            "use_sort": self.use_sort,
            "verbose": self.verbose,
            "n_clusters": self.n_clusters,
        }
            
        if self.qp is not None:
            result["qp"] = self.qp
        
        # handle attribute_codec_registry
        if self.attribute_codec_registry is not None:
            result["attribute_codec_registry"] = self.attribute_codec_registry.to_dict()
        
        return result

@dataclass
class Config: