Z mého pohledu si ten kód pro distinct v JS a hack se sum řešící flatMap v pythonu nemají co vyčítat, obojí je maximálně nepřehledné.
Mimochodem stačí se podívat, jaký je výkon toho hacku se sum
https://stackoverflow.com/a/45323085Funkce, která u jednoúrovňového číselného seznamu vrátí jedno číslo (jejich součet), u jednoúrovňového nečíselného seznamu hodí chybu, u víceúrovňového s libovolnými hodnotami sníží level o jeden, to opravdu nepovažuji za bezpečného kandidáta pro flatMap. Obzvláště když pythonní proměnná akceptuje cokoliv.