Fix lightgbm slow test (mlflow#13697)

Signed-off-by: Weichen Xu <[email protected]>

.github/workflows/slow-tests.yml

@@ -71,7 +71,8 @@ jobs:
           source ./dev/install-common-deps.sh --ml
           # transformers requires `tf-keras` when keras 3.0 is installed.
           # See https://github.com/huggingface/transformers/issues/27377 for more details.
-          python -m pip install langchain_experimental tf-keras
+          # Installing pyarrow < 18 to prevent lightgbm test pip installation resolution conflicts.
+          python -m pip install langchain_experimental tf-keras "pyarrow<18"
       - uses: ./.github/actions/show-versions
       - uses: ./.github/actions/pipdeptree
       - name: Run tests

tests/pyfunc/docker/conftest.py

@@ -78,6 +78,9 @@ def save_model_with_latest_mlflow_version(flavor, extra_pip_requirements=None, *
     else:
         extra_pip_requirements = extra_pip_requirements or []
         extra_pip_requirements.append(f"mlflow=={latest_mlflow_version}")
+        if flavor == "lightgbm":
+            # Adding pyarrow < 18 to prevent pip installation resolution conflicts.
+            extra_pip_requirements.append("pyarrow<18")
         kwargs["extra_pip_requirements"] = extra_pip_requirements
     flavor_module = getattr(mlflow, flavor)
     flavor_module.save_model(**kwargs)