use list

test_unet.py

@@ -94,7 +94,7 @@ def separate(
         frames = 2584
         min_frames = 516
         masked_tf_rep = []
-        chunks = mag.split(frames, dim=-1)
+        chunks = list(mag.split(frames, dim=-1))
         if len(chunks) > 1 and chunks[-1].shape[-1] < min_frames:
             chunks[-2:] = torch.cat(chunks[-2:], -1).chunk(2, -1)