mlflow-snowflake
mlflow-snowflake copied to clipboard
Allow explicit UDF arg & return types specifications
Two use cases:
- MLflow packaged models do not enforce the specification of column specs. Currently, we bail in this case. Would be good if customer could manually specify.
- MLflow auto schema capture could result in
numpy
type ofobject
which could be because of python list, dict or even str. There's no way we could generate proper UDF signature based onobject
type. (reported by @jonwiggins)