From 2a609ecf00ab4f6ee2264a19e267c3e4ff392694 Mon Sep 17 00:00:00 2001 From: Muhammad Anas Raza Date: Wed, 28 Feb 2024 16:29:39 -0500 Subject: [PATCH] changes in requirements --- requirements/requirements_test.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements/requirements_test.txt b/requirements/requirements_test.txt index a626de1..494f270 100644 --- a/requirements/requirements_test.txt +++ b/requirements/requirements_test.txt @@ -1,3 +1,4 @@ torch --index-url https://download.pytorch.org/whl/cpu +torchmetrics tensorflow jax[cpu] \ No newline at end of file