From 1ea8939fd18b0ddb9837f219ffc1351cec32f937 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Sun, 21 Jan 2024 22:46:08 -0800 Subject: [PATCH] bump --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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