Uncomment test cases.

PiperOrigin-RevId: 399893074
google-deepmind · Sep 30, 2021 · 3343496 · 3343496
1 parent 0363bc4
commit 3343496
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/distrax/_src/distributions/independent_test.py b/distrax/_src/distributions/independent_test.py
@@ -200,9 +200,9 @@ def test_log_prob(self, batch_ndims, value):
 
   @chex.all_variants(with_pmap=False)
   @parameterized.named_parameters(
-      # ('entropy, batch dims None', 'entropy', None),
-      # ('entropy, batch dims 1', 'entropy', 1),
-      # ('entropy, batch dims 3', 'entropy', 3),
+      ('entropy, batch dims None', 'entropy', None),
+      ('entropy, batch dims 1', 'entropy', 1),
+      ('entropy, batch dims 3', 'entropy', 3),
       ('mean, batch dims None', 'mean', None),
       ('mean, batch dims 1', 'mean', 1),
       ('mean, batch dims 3', 'mean', 3),