mlflow_provider.hooks.pyfunc

Module Contents

Classes

MLflowPyfuncHook

Hook that interacts with the mlflow.pyfunc module. https://www.mlflow.org/docs/latest/python_api/mlflow.pyfunc.html

class mlflow_provider.hooks.pyfunc.MLflowPyfuncHook(mlflow_conn_id)

Bases: mlflow_provider.hooks.base.MLflowBaseHook

Hook that interacts with the mlflow.pyfunc module. https://www.mlflow.org/docs/latest/python_api/mlflow.pyfunc.html

Parameters:

• aws_conn_id (str) – AWS connection to use with hook

• target_uri (str) – target system URI to deploy model to. (ie ‘sagemaker’)

hook_name = 'MLflow Pyfunc'

get_conn()

Returns MLflow Pyfunc Client.