No-op simulation used when compression simulation is disabled.
| 63 | |
| 64 | |
| 65 | class NullCompressionSimulation(CompressionSimulationBase): |
| 66 | """No-op simulation used when compression simulation is disabled.""" |
| 67 | |
| 68 | def __init__(self, device: Optional[torch.device] = None): |
| 69 | super().__init__(CompSimConfig(enabled=False), device=device) |
| 70 | |
| 71 | def run( |
| 72 | self, splats: Mapping[str, Tensor] | ParameterDict, step: int |
| 73 | ) -> SimulationResult: |
| 74 | tensor_dict = _maybe_to_tensor_dict(splats) |
| 75 | return SimulationResult(splats=dict(tensor_dict)) |
| 76 | |
| 77 | def step_optimizers(self, step: int) -> None: |
| 78 | return |
| 79 | |
| 80 | def state_dict(self) -> Dict[str, Any]: |
| 81 | return {} |
| 82 | |
| 83 | def load_state_dict(self, state: MutableMapping[str, Any]) -> None: |
| 84 | if state: |
| 85 | raise ValueError("NullCompressionSimulation received unexpected state during load") |
| 86 | |
| 87 | |
| 88 | class DefaultCompressionSimulation(CompressionSimulationBase): |
no outgoing calls