something wrong in bert_data.py?

[sloth2012]

in the get_data funciton, should variable cls be a list?

[sloth2012]

if that, cls2idx[cls] should be wrong.

the code is about in line 197 in modules.data.bert_data

[king-menin]

cls - is the label of class (usally str) or None if is_cls=False in get_data.
My bad is i didn't check any types of arguments.

PS. Maybe code should be simplify. But i think, that it works correctly (with right input data). If you have any suggestions of improving. Please, come on :).