diff --git a/pyproject.toml b/pyproject.toml index a60cc2d..d4e1a12 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -53,7 +53,7 @@ docs = [ ] notebooks = [ "scikit-learn>=1.3.2,<1.5", - "torch>=2.1.2,<2.3", + "torch>=2.1.2,<2.4", "torchvision>=0.16.2,<0.19", "ipywidgets>=8.1.1,<8.2", ]