Searching for message passing depth

[nutellamok]

Dear authors,
could you point out where in the released code you implement Algorithm 1 in the paper (optimal message passing depth search)?

[OneZeroer]

Dear authors
When I run the code , I get "ImportError: cannot import name 'TransInput'" . But I can not slove the problem by myself . I wander that if I can get your help.
My qq : 542086695
I'm a student from BUAA , I run the code for a competition.