(self, **kwargs)
| 80 | return recon |
| 81 | |
| 82 | def sample(self, **kwargs) -> torch.Tensor: |
| 83 | cond_orders: torch.Tensor = kwargs['cond_orders'] |
| 84 | x_0: torch.Tensor = kwargs['x'] |
| 85 | cond_lob: torch.Tensor = kwargs['cond_lob'] |
| 86 | x_0, cond_orders = self.type_embedding(x_0, cond_orders) |
| 87 | x_0 = torch.zeros_like(x_0) |
| 88 | weights = self.sampler.weights() |
| 89 | x_t = self.diffuser.sample(x_0, cond_orders, cond_lob, weights) |
| 90 | return x_t |
| 91 | |
| 92 | |
| 93 | def single_step(self, cond_orders, x_0, cond_lob, batch_idx=None): |
no test coverage detected