search

wengong-jin / abdockgen

MIT License
104 stars 22 forks source link

[BUG] The parameter 'topk' should transform as interger #6

Open Autumn-Roy opened 11 months ago

Autumn-Roy commented 11 months ago

In the file generate.py.

topk = sys.argv[3] will get a string type value. But in line for cdr,ppl in new_res[:topk]:, the 'topk' should be a int type. So it should add an transform line to change the value type. This is my way:

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