Support non-TRANSFORM model Predictors

Would it be possible to make a tensorflow predictor that did not expect the model to be queried with the TRANSFORM command, but rather the PREDICT command? This library is great for locally using exported models from BQML, but I am manually having to use the Predictor._get_transform_result() function to bypass 2-stage TRANSFORM+PREDICT logic.

Thanks!