Thanks for your code! I was trying to use the deep biaffine model for DEP and a type error popped up:  Could you tell me what's wrong? Thank you!