mlflow_provider.operators.pyfunc

Module Contents

Classes

ModelLoadAndPredictOperator

Deploy MLflow models
class mlflow_provider.operators.pyfunc.ModelLoadAndPredictOperator(*, mlflow_conn_id='mlflow_default', model_uri, suppress_warnings=False, dst_path=None, data, **kwargs)

Bases: airflow.operators.python._BasePythonVirtualenvOperator

Deploy MLflow models

Parameters:

  • name (str) – Unique name to use for deployment

  • model_uri (str) – URI of MLflow model

  • target_uri (str) – URI of location to deploy the model (ie ‘sagemaker’)

  • target_conn_id (str) – Airflow connection id for target system

  • flavor (str) – Model flavor to deploy. If unspecified, a default flavor will be chosen.

  • config (dict) – Target-specific configuration for the deployment

  • endpoint (str) – Endpoint to create the deployment under. May not be supported by all targets

template_fields: Sequence[str] = ('model_uri', 'dst_path', 'data', 'op_args', 'op_kwargs')
template_fields_renderers
template_ext = ()
ui_color = '#f4a460'
execute(context)