Update test_svd.py

Signed-off-by: Karthik Uppuluri <karthik.uppuluri@fmr.com>
fidelity · Dec 4, 2024 · acb6e58 · acb6e58
1 parent 7de9ff0
commit acb6e58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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):