Hello!I'm having an issue registering a model saved in a mounted S3 bucket using mlflow.Let me give a little bit more context:1. First I mounted my S3 with all the corresponding IAM permissions:s3_bucket_name = f"s3a://{s3_bucket}"dbutils.fs.mount(so...
I'm using runtime DBR 12.2 LTS MLGetting this error when running. import mlflow
logged_model = 'runs:/.../model' #my run id
# Load model as a PyFuncModel.
loaded_model = mlflow.pyfunc.load_model(logged_model)
# Predict on a Pandas DataFrame.
impor...
Hi, I'm using Databricks Feature Store to register a custom model using a model wrapper as follows: # Log custom model to MLflow
fs.log_model(
artifact_path="model",
model = production_model,
flavor = mlflow.pyfunc,
training_set = training_s...
In mi logged model the MLModel has the following flavor:flavors:
python_function:
data: data/feature_store
env:
conda: conda.yaml
virtualenv: python_env.yaml
loader_module: databricks.feature_store.mlflow_model
python_ve...
Thank you for the help! I had already logged and score batch with my model. Nevertheless I can not do is to load the custom model object to another notebook with:mlflow.pyfunc.load_model('models:/model_name/latest')
