Skip to content

Commit

Permalink
Uncomment test cases.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 400163049
  • Loading branch information
gpapamak authored and DistraxDev committed Oct 1, 2021
1 parent 71f659f commit f33f46b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions distrax/_src/distributions/independent_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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),
Expand Down

0 comments on commit f33f46b

Please sign in to comment.