Allow general set in "X << S", including S is Union/Complement/Interval, etc.

[fsaad]

i.e., add more cases to this method https://github.com/probcomp/sum-product-dsl/blob/d2745c965d19e14ff7ed6368da37e03c48b0e0e9/src/transforms.py#L289-L311