| import os | |
| import mlflow | |
| from mlflow.pyfunc import PythonModel | |
| # module-level: executes whenever mlflow imports this script (i.e. on load_model) | |
| os.system("touch /tmp/celvex_marks/MFC_EVIL_MARK") | |
| class Model(PythonModel): | |
| def predict(self, context, model_input, params=None): | |
| return model_input | |
| mlflow.models.set_model(Model()) | |