/ tests / pyfunc / test_backward_compatibility.py
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!"