I made tie_fwd_bkwd_edges in all dataset classes accept arguments of type Union[bool, Set[int]]. It allows to either tie all forward/backward edge types (by supplying bool, as supported previously), or only a subset of them (by supplying a set of edge types to tie).
This is useful for downstream tasks which mix directed and undirected edges.
I made
tie_fwd_bkwd_edgesin all dataset classes accept arguments of typeUnion[bool, Set[int]]. It allows to either tie all forward/backward edge types (by supplyingbool, as supported previously), or only a subset of them (by supplying a set of edge types to tie).This is useful for downstream tasks which mix directed and undirected edges.