Thank you for sharing the code!
Does the code match the PyTransformer main branch version?
It throws an error at runtime, saying that the number of arguments in the transformer's _build_graph function is incorrect.