| 3 | |
| 4 | |
| 5 | class SampleClassBase(abc.ABC): |
| 6 | def sample(self, logits, x_0): |
| 7 | raise NotImplementedError |
| 8 | |
| 9 | def post_process_sample_in_prediction(self, sample, x_0): |
| 10 | return sample |
| 11 | |
| 12 | |
| 13 | class Categorical(SampleClassBase): |
nothing calls this directly
no outgoing calls
no test coverage detected