chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,10 @@
|
||||
from mlflow.models import ModelConfig, set_model
|
||||
|
||||
|
||||
def predict(model_input: list[str]):
|
||||
model_config = ModelConfig(development_config="tests/pyfunc/sample_code/config.yml")
|
||||
timeout = model_config.get("timeout")
|
||||
return f"This was the input: {model_input[0]}, timeout {timeout}"
|
||||
|
||||
|
||||
set_model(predict)
|
||||
Reference in New Issue
Block a user