Fix the issue of "UnboundLocalError: local variable 'train_prots' referenced before assignment" when train_list and test_list are provided.
Fix the issue of "UnboundLocalError: local variable 'train_prots' referenced before assignment" when train_list and test_list are provided.