[python] add export def
[atlaas] cleanup includes and move length_sq