USING EXTERNAL FUNCTION grid_disk(origin BIGINT, k INT) 
RETURNS ARRAY(BIGINT) 
LAMBDA '<ARN>'
SELECT grid_disk(581975902628347903, 1) AS k1, grid_disk(581975902628347903, 3) AS k3

