bug fix

Signed-off-by: Pratyush Muthukumar <[email protected]>
xuanzic · Feb 22, 2024 · fdd5b5b · fdd5b5b
1 parent d39e15f
commit fdd5b5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/multimodal/data/neva/neva_dataset.py b/nemo/collections/multimodal/data/neva/neva_dataset.py
@@ -166,7 +166,7 @@ def flatten_frames(self, cap):
             else:
                 partition_size = max(1, len(cap) // self.data_cfg['num_frames'])
                 for i in range(0, len(cap), partition_size):
-                    frame = cap[i].asnumpy()[:, :, ::-1]
+                    rgb_frame = cap[i].asnumpy()[:, :, ::-1]
                     img = Image.fromarray(rgb_frame).convert('RGB')
                     frames.append(img)
                 return frames