UNION within CTE causes error

[kevinkelleher12]

Thank you for developing this library! I'm getting an error when using UNION within a CTE:

CREATE TABLE OUT AS
WITH CTE AS (
SELECT COL1, COL2
FROM IN1
UNION ALL
SELECT COL1, COL2
FROM IN2
)
SELECT COL1, COL2
FROM CTE
;
'''
result = LineageRunner(sql, dialect='ansi')
print(result)

I'm wondering if when get_subqueries calls segment.get_child("set_expression") if this should be some kind of recursive search, but not sure.

[reata]

Yes this causes internal failure. If we can make this recursive search it's all the better. Or we can just patch it with a special logic to handle segment.type == "with_compound_statement' case. Are you interested in a PR? I think you're very close to solving it.