diff --git a/pyproject.toml b/pyproject.toml index 397a880..5ca0af5 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -84,7 +84,7 @@ ml = [ "scikit-learn", "imbalanced-learn", "xgboost", - "torch==2.2.2", + "torch==2.7.0", ] [project.entry-points]