reword keyword arg handling in theano_function