Skip to content

Commit

Permalink
Fix scikit-learn package
Browse files Browse the repository at this point in the history
  • Loading branch information
@cyugao
cyugao authored Dec 2, 2022
1 parent 69640d0 commit 17e3615
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion requirements-dev.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ pynvml == 8.0.4
numpy == 1.22.0

# For layerwise gradient scaler
sklearn >= 0.0
scikit-learn >= 0.0

This comment has been minimized.

Sign in to view

Copy link
@min-xu-ai

min-xu-ai Dec 2, 2022

Contributor

or just fix the version of sklearn == 0.0 to avoid updating this package.

Also, the comment on line 31 is stale. Can you help remove that?

This comment has been minimized.

Sign in to view

Copy link
@cyugao

cyugao Dec 3, 2022

Author Contributor

Will remove the comment in the next commit.

# For weigit. These are actually user requirements, not developer requirements.
# However, due to the experimental nature of weigit, we don't expose to the
Expand Down

0 comments on commit 17e3615

Please sign in to comment.