(self, x, sigma_min, sigma_max, seed=None, transform=lambda x: x)
| 105 | """ |
| 106 | |
| 107 | def __init__(self, x, sigma_min, sigma_max, seed=None, transform=lambda x: x): |
| 108 | self.transform = transform |
| 109 | t0, t1 = self.transform(torch.as_tensor(sigma_min)), self.transform(torch.as_tensor(sigma_max)) |
| 110 | self.tree = BatchedBrownianTree(x, t0, t1, seed) |
| 111 | |
| 112 | def __call__(self, sigma, sigma_next): |
| 113 | t0, t1 = self.transform(torch.as_tensor(sigma)), self.transform(torch.as_tensor(sigma_next)) |
nothing calls this directly
no test coverage detected