bikes.io.pyfunc_model

Model code that MLflow runs to rebuild the PyFunc adapter of project models.

MLflow copies this file next to each model logged by CustomSaver and executes it on load, so the adapter is defined by code instead of being pickled.

https://mlflow.org/docs/latest/ml/model/models-from-code/

 1"""Model code that MLflow runs to rebuild the PyFunc adapter of project models.
 2
 3MLflow copies this file next to each model logged by `CustomSaver` and executes it on
 4load, so the adapter is defined by code instead of being pickled.
 5
 6https://mlflow.org/docs/latest/ml/model/models-from-code/
 7"""
 8
 9# %% IMPORTS
10
11import mlflow
12
13from bikes.io import registries
14
15# %% MODELS
16
17ADAPTER = registries.CustomSaver.Adapter()
18
19mlflow.models.set_model(ADAPTER)
ADAPTER = <bikes.io.registries.CustomSaver.Adapter object>