Yes, we can face this issue locally by explicitly requiring the inputs for `dim_handling` inside the header for `theano_function`.  In this way the only non-explicit keyword arguments should be passed down to `theano.function`.  Does this seem reasonable?