When running the code with a non-bipartite graph I have encountered this error for line 81 of the data_preprocessing.py file.
This can be solved by replacing the line with
new_test_node_set = set(random.sample(sorted(test_node_set), int(0.1 * n_total_unique_nodes)))
Hope this helps