Skip to content

Commit

Permalink
Fixing dqn test
Browse files Browse the repository at this point in the history
  • Loading branch information
boris-il-forte committed Dec 25, 2022
1 parent 1360869 commit 0359512
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/algorithms/test_dqn.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ def test_dqn_logger(tmpdir):
loss_file = np.load(logger.path / 'loss_Q.npy')

assert loss_file.shape == (90,)
assert loss_file[0] == 0.976540744304657 and loss_file[-1] == 0.6936991810798645
assert loss_file[0] == 0.9765409231185913 and loss_file[-1] == 0.6936992406845093


def test_prioritized_dqn():
Expand Down

0 comments on commit 0359512

Please sign in to comment.