saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-28 20:11:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove useless entropy testing

Browse source
This commit is contained in:
Quentin Gallouédec 2022-11-29 13:10:54 +01:00
parent 2065b7ae60
commit d39197c555
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_distributions.py
View file

@ -78,7 +78,6 @@ def test_get_distribution(dummy_model_distribution_obs_and_actions):
log_prob_2 = distribution.log_prob(actions)
entropy_2 = distribution.entropy()
assert entropy_1 is not None
assert entropy_2 is not None
assert th.allclose(log_prob_1, log_prob_2)
assert th.allclose(entropy_1, entropy_2)
@ -230,3 +229,6 @@ def test_kl_divergence(dist_type):
)
assert th.allclose(full_kl_div, ad_hoc_kl)
test_kl_divergence(StateDependentNoiseDistribution(N_ACTIONS).proba_distribution(
th.rand(N_ACTIONS), th.rand([N_ACTIONS, N_ACTIONS]), th.rand([N_ACTIONS, N_ACTIONS])
))