+ result = []
+ N = 100
+ for _ in tqdm.tqdm(
+ range(0, nb + N, N), dynamic_ncols=True, desc="world generation"
+ ):
+ result.append(self.generate_frame_sequences_hard(100))
+ return torch.cat(result, dim=0)[:nb]
+
+ def generate_frame_sequences_hard(self, nb):