【Hackathon 6th No.5】Chi2 / LKJCholesky API

Author: cmcamdy

test/distribution/test_distribution_lkj_cholesky.py

@@ -61,7 +61,7 @@ def test_cvine_sample_shape(self):
         self._test_sample_shape_dim(sample_method)
 
     def _test_sample_shape_dim(self, sample_method):
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self._test_sample_shape(dim, sample_method)
 
     def _test_sample_shape(self, dim, sample_method):
@@ -117,7 +117,7 @@ def test_cvine_sample_shape(self):
         self._test_sample_shape_dim(sample_method)
 
     def _test_sample_shape_dim(self, sample_method):
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self._test_sample_shape(dim, sample_method)
 
     def _test_sample_shape(self, dim, sample_method):
@@ -140,7 +140,7 @@ def _test_sample_shape(self, dim, sample_method):
 class TestLKJCholeskyLogProb(unittest.TestCase):
     def test_log_prob_onion(self):
         sample_method = 'onion'
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self.dim = dim
             self._paddle_lkj_cholesky = lkj_cholesky.LKJCholesky(
                 dim, [1.0], sample_method
@@ -149,7 +149,7 @@ def test_log_prob_onion(self):
 
     def test_log_prob_cvine(self):
         sample_method = 'cvine'
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self.dim = dim
             self._paddle_lkj_cholesky = lkj_cholesky.LKJCholesky(
                 dim, [1.0], sample_method

test/distribution/test_distribution_lkj_cholesky_static.py

@@ -76,7 +76,7 @@ def test_cvine_sample_shape(self):
         self._test_sample_shape_dim(sample_method)
 
     def _test_sample_shape_dim(self, sample_method):
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self._test_sample_shape(dim, sample_method)
 
     def _test_sample_shape(self, dim, sample_method):
@@ -152,7 +152,7 @@ def test_cvine_sample_shape(self):
         self._test_sample_shape_dim(sample_method)
 
     def _test_sample_shape_dim(self, sample_method):
-        for dim in range(2, 5):
+        for dim in range(2, 4):
             self._test_sample_shape(dim, sample_method)
 
     def _test_sample_shape(self, dim, sample_method):