(self, x_0, real_cond_orders, real_cond_lob, weights)
| 82 | |
| 83 | |
| 84 | def sample(self, x_0, real_cond_orders, real_cond_lob, weights): |
| 85 | if self.sampling_type == "DDIM": |
| 86 | return self.ddim_sample(x_0, real_cond_orders, real_cond_lob) |
| 87 | elif self.sampling_type == "DDPM": |
| 88 | return self.ddpm_sample(x_0, real_cond_orders, real_cond_lob, weights) |
| 89 | |
| 90 | |
| 91 | def ddim_sample(self, x_0, cond_orders, cond_lob): |
no test coverage detected