test_backward_compatibility.py
1 import pytest 2 3 import mlflow 4 5 6 @pytest.mark.parametrize("version", ["2.7.1", "2.8.1"]) 7 def test_backward_compatibility(version): 8 model = mlflow.pyfunc.load_model(f"tests/resources/pyfunc_models/{version}") 9 assert model.predict("MLflow is great!") == "MLflow is great!"