Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update AutoGluon to use uv during setup.sh #643

Merged
merged 1 commit into from
Nov 2, 2024

Conversation

Innixma
Copy link
Collaborator

@Innixma Innixma commented Oct 29, 2024

  • Update AutoGluon to use uv instead of pip for installation in setup.sh This aligns with mainline AutoGluon now switching to uv for its install instructions.

  • The switch to uv reduces AutoGluon installation time from ~10 minutes to ~40 seconds on EC2. This also avoids a bug in pip where it fails to install AutoGluon.

  • Removed setuptools and wheel installs, as they aren't needed by uv and there is a bug in torch 1.4.x that causes an exception if setuptools is installed at the same time.

  • Will now install from source in non-editable mode. Previously it would install in editable mode, which could cause bugs if running AMLB locally alongside a pre-existing local AutoGluon install. This is now avoided by ensuring that the venv directly uses the code pulled in setup.sh.

Copy link
Collaborator

@PGijsbers PGijsbers left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sorry for the delay. I wanted to try it out locally to understand where uv installs its packages if created from a virtual environment to ensure the normal cleanup instructions are still valid. It behaves as I expected so we're good to go :)

@PGijsbers PGijsbers merged commit 4c4f35d into openml:master Nov 2, 2024
10 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants