Hi, I get `ValueError: Cannot take the length of Shape with unknown rank` on this line https://github.com/GraphSAINT/GraphSAINT/blob/a051742ff2de4094c97eb523d7108a4fc1d22739/graphsaint/tensorflow_version/model.py#L128 I got around it by commenting out the if check and executing the body of the clause. Have you seen this before? Thanks.