(self, exc_type, exc, tb)
| 588 | return self |
| 589 | |
| 590 | def __exit__(self, exc_type, exc, tb) -> None: |
| 591 | if self._ctx is not None: |
| 592 | self._ctx.__exit__(exc_type, exc, tb) |
| 593 | self._ctx = None |
| 594 | |
| 595 | @torch.inference_mode() |
| 596 | def decode_codes(self, codes: torch.LongTensor) -> torch.Tensor: |
nothing calls this directly
no outgoing calls
no test coverage detected