microsoft / tf2-gnn

TensorFlow 2 library implementing Graph Neural Networks
MIT License
369 stars 73 forks source link

Fix the type of tie_fwd_bkwd_edges #21

Closed kmaziarz closed 4 years ago

kmaziarz commented 4 years ago

The type of tie_fwd_bkwd_edges used to be Union[bool, Set[int]], but JSON doesn't support sets, so this cannot be supplied on the command line, nor serialized into a JSON file. Hence, I change Set to List, and convert the parameter value into a set internally.