diff --git a/setup.py b/setup.py index 0dbebab108..87a3f9a852 100644 --- a/setup.py +++ b/setup.py @@ -76,7 +76,7 @@ def package_files(prefix: str, directory: str, extension: str): install_requires = [ 'pyyaml>=6.0,<7', 'tqdm>=4.62.3,<5', - 'torchmetrics>=0.10.0,<1.1', + 'torchmetrics>=0.10.0,<1.3.1', 'torch_optimizer>=0.3.0,<0.4', 'torchvision>=0.13.1,<0.20', # TODO: Tighten before release 'torch>=1.13.1,<2.3.1', # TODO: Tighten before release