Some undefined variables and classes in tkgc.py

I'm interested in your work, but the code seems incomplete. I can't find the definitions of class "TripleDataset" and variable "task_name" in tkgc.py, and there is no import or reference for align.py and funtional.py. Could you please update the code?
Besides, where can I get the datasets file "train.csv", "validation.csv" and "test.csv"?

have you solved it? thank you