Bugfix - was not picking up @OptimizeByInference annotation for higher order functions (i.e. linear models)