From f25cb0b3c0255f91248065d90836b74bf1c8d9b1 Mon Sep 17 00:00:00 2001 From: Tom Vercauteren Date: Wed, 30 Nov 2022 10:52:00 +0000 Subject: [PATCH] specify minimum torch version (fixes #11 ) + remove tests from pip install as this created mess (see #18) --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 2a04615..f44b7c9 100644 --- a/setup.py +++ b/setup.py @@ -22,9 +22,9 @@ def readme(): author="CAI4CAI research group", author_email="contact@cai4cai.uk", license="Apache-2.0", - packages=setuptools.find_packages(), + packages=setuptools.find_packages(exclude=("tests",)), install_requires=[ - "torch", + "torch>=1.13", ], extras_require={ "extras": ["jax", "cupy"],