Example generation is not running

[serbulent-av]

Hello,

The generation example produce error as;

python generate.py ckpts/HERN_gen.ckpt data/rabd/test_data.jsonl 1 > results/HERN.txt
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 58/58 [00:01<00:00, 49.16it/s]
0%| | 0/58 [00:24<?, ?it/s]
Traceback (most recent call last):
File "/home/serbulent_antiverse_io/code/abdockgen/generate.py", line 60, in
for cdr,ppl in new_res[:topk]:
TypeError: slice indices must be integers or None or have an index method

[Autumn-Roy]

This error is the type error, so I add the type transformer in the generate.py. This is my solution:

 topk = sys.argv[3]
    if not isinstance(topk, int):
        topk = int(topk)