diff --git a/setup.py b/setup.py index 2765b07..dd3739d 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ long_description = fh.read() extras = {} -extras["torch"] = ["torch>=1.5,<2.2"] +extras["torch"] = ["torch>=1.5,<2.3"] extras["all"] = extras["torch"] extras["docs"] = ["mkdocs-material"]