Adjust tests and fix bug introduced in previous commit

Author: alexgdebeer

deep_tensor/irt/sirt.py

@@ -207,9 +207,9 @@ def _marginalise_backward(self) -> None:
 
     def _eval_rt_local_forward(self, ls: Tensor) -> Tensor:
 
-        dim_ls = ls.shape[1]
+        num_ls, dim_ls = ls.shape
         zs = torch.zeros_like(ls)
-        Gs_prod = torch.ones_like(ls[:, 0])
+        Gs_prod = torch.ones((num_ls, 1), device=ls.device)
 
         cores = self.ftt.cores
         Bs = self._Bs_f 

tests/test_domains/test_bounded_domain.py

@@ -11,8 +11,8 @@
 class TestLinearDomain(unittest.TestCase):
 
     def setup_domain(self):
-        bounds = torch.tensor([-2.0, 4.0])
-        domain = dt.BoundedDomain(bounds=bounds)
+        bounds = [-2.0, 4.0]
+        domain = dt.BoundedDomain(bounds)
         return domain
 
     def test_linear_domain(self):
@@ -22,8 +22,9 @@ def test_linear_domain(self):
         domain = self.setup_domain()
 
         bounds_true = torch.tensor([-2.0, 4.0])
+        bounds = torch.tensor(domain.bounds)
 
-        self.assertTrue((domain.bounds - bounds_true).abs().max() < 1e-8)
+        self.assertTrue((bounds - bounds_true).abs().max() < 1e-8)
         self.assertAlmostEqual(domain.dxdl, 3.)
         self.assertAlmostEqual(domain.mean, 1.)
         self.assertAlmostEqual(domain.left, -2.)

tests/test_polynomials/test_cdf.py

@@ -35,7 +35,7 @@ class are inverses of one another.
         for poly in polys:
             with self.subTest(poly=poly):
 
-                cdf = dt.construct_cdf(polys[poly])
+                cdf = dt.construct_cdf(polys[poly], error_tol=1e-10)
 
                 ls = torch.linspace(-1.0, 1.0, n_ls)
                 ps = dummy_pdf(cdf.nodes) + 1e-2

tests/test_polynomials/test_piecewise_cdf.py

@@ -12,7 +12,7 @@ class TestPiecewiseCDF(unittest.TestCase):
 
     def setup_cdf(self):
         poly = dt.Lagrange1(num_elems=2)
-        cdf = dt.Lagrange1CDF(poly=poly)
+        cdf = dt.Lagrange1CDF(poly=poly, error_tol=1e-10)
         return cdf
 
     def test_lagrange_1d_cdf(self):