Raise a ValueError（invalid literal for int() with base 10: 'tensor(589,'）on the step 2 of generate

[WangCEChen]

Hi @ratishsp

Traceback (most recent call last):
File "scripts/create_content_plan_from_index.py", line 29, in
output.append(input[int(record)].encode("utf-8"))
ValueError: invalid literal for int() with base 10: 'tensor(589,'

The first line of file which is generated from step 1:
'tensor(589, device='cuda:0') tensor(590, device='cuda:0') tensor(582, device='cuda:0') tensor(420, device='cuda:0') tensor(421, device='cuda:0') tensor(402, device='cuda:0') tensor(414, device='cuda:0') tensor(415, device='cuda:0') tensor(376, device='cuda:0') tensor(377, device='cuda:0') tensor(358, device='cuda:0') tensor(370, device='cuda:0') tensor(371, device='cuda:0') tensor(373, device='cuda:0') tensor(332, device='cuda:0') tensor(333, device='cuda:0') tensor(314, device='cuda:0') tensor(326, device='cuda:0') tensor(327, device='cuda:0') tensor(156, device='cuda:0') tensor(157, device='cuda:0') tensor(138, device='cuda:0') tensor(150, device='cuda:0') tensor(90, device='cuda:0') tensor(91, device='cuda:0') tensor(72, device='cuda:0') tensor(84, device='cuda:0') tensor(85, device='cuda:0') tensor(68, device='cuda:0') tensor(69, device='cuda:0') tensor(50, device='cuda:0') tensor(62, device='cuda:0') tensor(66, device='cuda:0') tensor(24, device='cuda:0') tensor(25, device='cuda:0') tensor(6, device='cuda:0') tensor(18, device='cuda:0') tensor(19, device='cuda:0')
'
Am I running something wrong? Could you please take a look at it for me?