Function with table as argument that return a table

Hi,

I want to create a Python function that accept a query as argument, do some processing and return another table, In addition to the table columns I want to have some additional scalar parameters and i am getting the following error: function requires a single sub query.

CREATE function f1(col1 string, col2 int, col3 double, scalar1 int) RETURNS TABLE(col1 string, col2 double, col3 double, col4 double) LANGUAGE PYTHON

Is this the normal behavior of Table function, as in aggregate function I can combine columns with scalar parameters, any workaround for this issue.

Thank you.