I met problems when running my own dataset

[LaoWang-123]

My dataset has 6 types of edge and 3 types of nodes. When I run this model with main.py without schema, it shows the failure as follows.

Traceback (most recent call last):
File "src\main.py", line 226, in
average_auc, average_f1, average_pr = train_model(training_data_by_type, feature_dic, log_name + '_' + time.strftime('%Y-%m-%d %H-%M-%S',time.localtime(time.time())))
File "src\main.py", line 181, in train_model
tmp_auc, tmp_f1, tmp_pr = evaluate(final_model[edge_types[i]], valid_true_data_by_edge[edge_types[i]], valid_false_data_by_edge[edge_types[i]])
KeyError: '5'
keyerror is an edge_type name

However, when I use schema, I could see the generate walks work on 6 layers while other funcitons (save walks,count vocab, generate train pairs) only work on three layers.
Thank you very much.