From acb6e58bd22bfb9eb00e78ac15849bd81bafcb26 Mon Sep 17 00:00:00 2001 From: Karthik Uppuluri Date: Wed, 4 Dec 2024 16:01:53 -0500 Subject: [PATCH] Update test_svd.py Signed-off-by: Karthik Uppuluri --- tests/test_svd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_svd.py b/tests/test_svd.py index 64bd98b..e542431 100644 --- a/tests/test_svd.py +++ b/tests/test_svd.py @@ -13,9 +13,9 @@ def test_fit_transform(self): expected = torch.tensor([[0.8526761532, -0.5070778131], [0.9837458134, -0.0636523664], [0.7350711226, 0.6733918786]], dtype=torch.float32) - self._test_fit_transform(tw, expected) + self._test_fit_transform(tw, expected, svd=True) self._reset_seed() - self._test_fit_before_transform(tw, expected) + self._test_fit_before_transform(tw, expected, svd=True) def test_min_components(self): with self.assertRaises(ValueError):