Skip to content

Remove numpy<2 pin for pmdarima compatibility#18933

Merged
harupy merged 6 commits intomasterfrom
copilot/remove-numpy-pin
Nov 20, 2025
Merged

Remove numpy<2 pin for pmdarima compatibility#18933
harupy merged 6 commits intomasterfrom
copilot/remove-numpy-pin

Conversation

Copy link
Contributor

Copilot AI commented Nov 20, 2025

Related Issues/PRs

Addresses pmdarima issue alkaline-ml/pmdarima#577 which has been fixed upstream in pmdarima >= 2.1.0.

What changes are proposed in this pull request?

The numpy<2 pin was added in PR #18566 as a workaround for pmdarima incompatibility with numpy 2.x. Since the upstream issue is resolved in pmdarima >= 2.1.0, the blanket pin is no longer necessary.

Changes:

  • Remove pip install "numpy<2" from .github/workflows/master.yml python job
  • Add version-specific constraint "< 2.1.0": ["numpy<2"] to pmdarima requirements in mlflow/ml-package-versions.yml
  • Increment cache key N from 0 to 1 in .github/actions/py-cache-key/action.yml to invalidate pip cache and ensure numpy 2.x is used

This ensures that pmdarima versions < 2.1.0 (including the currently tested 2.0.4) still get numpy<2 as required, while future versions >= 2.1.0 won't have the numpy<2 constraint. The cache key increment forces a cache refresh so CI will use numpy 2.x after the pin is removed.

How is this PR tested?

  • Existing unit/integration tests
  • New unit/integration tests
  • Manual tests

Does this PR require documentation update?

  • No. You can skip the rest of this section.
  • Yes. I've updated:
    • Examples
    • API references
    • Instructions

Release Notes

Is this a user-facing change?

  • No. You can skip the rest of this section.
  • Yes. Give a description of this change to be included in the release notes for MLflow users.

What component(s), interfaces, languages, and integrations does this PR affect?

Components

  • area/tracking: Tracking Service, tracking client APIs, autologging
  • area/models: MLmodel format, model serialization/deserialization, flavors
  • area/model-registry: Model Registry service, APIs, and the fluent client calls for Model Registry
  • area/scoring: MLflow Model server, model deployment tools, Spark UDFs
  • area/evaluation: MLflow model evaluation features, evaluation metrics, and evaluation workflows
  • area/gateway: MLflow AI Gateway client APIs, server, and third-party integrations
  • area/prompts: MLflow prompt engineering features, prompt templates, and prompt management
  • area/tracing: MLflow Tracing features, tracing APIs, and LLM tracing functionality
  • area/projects: MLproject format, project running backends
  • area/uiux: Front-end, user experience, plotting, JavaScript, JavaScript dev server
  • area/build: Build and test infrastructure for MLflow
  • area/docs: MLflow documentation pages

How should the PR be classified in the release notes? Choose one:

  • rn/none - No description will be included. The PR will be mentioned only by the PR number in the "Small Bugfixes and Documentation Updates" section
  • rn/breaking-change - The PR will be mentioned in the "Breaking Changes" section
  • rn/feature - A new user-facing feature worth mentioning in the release notes
  • rn/bug-fix - A user-facing bug fix worth mentioning in the release notes
  • rn/documentation - A user-facing documentation change worth mentioning in the release notes

Should this PR be included in the next patch release?

  • Yes (this PR will be cherry-picked and included in the next patch release)
  • No (this PR will be included in the next minor release)
Original prompt

Remove numpy pin in python job in master.yml. #18566 added it but (alkaline-ml/pmdarima#577 has been fixed


✨ Let Copilot coding agent set things up for you — coding agent works faster and does higher quality work when set up for your repo.

@harupy harupy marked this pull request as ready for review November 20, 2025 06:46
@github-actions
Copy link
Contributor

@Copilot Thank you for the contribution! Could you fix the following issue(s)?

⚠ Invalid PR template

This PR does not appear to have been filed using the MLflow PR template. Please copy the PR template from here and fill it out.

The pmdarima issue (alkaline-ml/pmdarima#577) has been fixed, so we can remove the numpy<2 pin that was added in PR #18566.

Changes:
- Removed numpy<2 pin from .github/workflows/master.yml (python job)
- Removed numpy<2 pin from mlflow/ml-package-versions.yml (pmdarima requirements)

Co-authored-by: harupy <17039389+harupy@users.noreply.github.com>
Copilot AI changed the title [WIP] Remove numpy pin in master.yml Remove numpy<2 pin for pmdarima compatibility Nov 20, 2025
Copilot AI requested a review from harupy November 20, 2025 06:51
Per feedback, pmdarima versions below 2.1.0 still require numpy<2, so adding version-specific constraint.

Co-authored-by: harupy <17039389+harupy@users.noreply.github.com>
Copilot AI requested a review from harupy November 20, 2025 06:57
@github-actions github-actions bot added area/build Build and test infrastructure for MLflow rn/none List under Small Changes in Changelogs. labels Nov 20, 2025
Incrementing N from 0 to 1 in py-cache-key action to force cache refresh and ensure numpy 2.x is used after removing the numpy<2 pin.

Co-authored-by: harupy <17039389+harupy@users.noreply.github.com>
Copilot AI requested a review from harupy November 20, 2025 07:10
harupy and others added 2 commits November 20, 2025 16:12
Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
@github-actions
Copy link
Contributor

Documentation preview for 49a8e6f is available at:

More info
  • Ignore this comment if this PR does not change the documentation.
  • The preview is updated when a new commit is pushed to this PR.
  • This comment was created by this workflow run.
  • The documentation was built by this workflow run.

@harupy harupy enabled auto-merge November 20, 2025 08:09
@harupy harupy added this pull request to the merge queue Nov 20, 2025
Merged via the queue into master with commit 95d4fa9 Nov 20, 2025
46 of 48 checks passed
@harupy harupy deleted the copilot/remove-numpy-pin branch November 20, 2025 08:13
Tian-Sky-Lan pushed a commit to Tian-Sky-Lan/mlflow that referenced this pull request Nov 24, 2025
Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: harupy <17039389+harupy@users.noreply.github.com>
Co-authored-by: Harutaka Kawamura <hkawamura0130@gmail.com>
Signed-off-by: Tian Lan <sky.blue266000@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/build Build and test infrastructure for MLflow rn/none List under Small Changes in Changelogs.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants