Skip to content

Commit 538f6f6

Browse files
Fix failing tests
1 parent d4cff0d commit 538f6f6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test_mdp_playground.py

+4
Original file line numberDiff line numberDiff line change
@@ -1313,6 +1313,7 @@ def test_discrete_rewardable_sequences(self):
13131313
config["repeats_in_sequences"] = False
13141314
config["delay"] = 0
13151315
config["sequence_length"] = 3
1316+
config["reward_every_n_steps"] = False
13161317
config["reward_scale"] = 1.0
13171318

13181319
config["generate_random_mdp"] = True
@@ -1475,6 +1476,7 @@ def test_discrete_multiple_meta_features(self):
14751476
config["repeats_in_sequences"] = False
14761477
config["delay"] = 1
14771478
config["sequence_length"] = 3
1479+
config["reward_every_n_steps"] = False
14781480
config["reward_scale"] = 2.5
14791481
config["reward_shift"] = -1.75
14801482
# config["transition_noise"] = 0.1
@@ -1733,6 +1735,7 @@ def test_discrete_image_representations(self):
17331735
config["repeats_in_sequences"] = False
17341736
config["delay"] = 1
17351737
config["sequence_length"] = 3
1738+
config["reward_every_n_steps"] = False
17361739
config["reward_scale"] = 2.5
17371740
config["reward_shift"] = -1.75
17381741
# config["transition_noise"] = 0.1
@@ -2196,6 +2199,7 @@ def test_discrete_diameter(self):
21962199
config["delay"] = 0
21972200
config["diameter"] = 3
21982201
config["sequence_length"] = 3
2202+
config["reward_every_n_steps"] = False
21992203
config["reward_scale"] = 1.0
22002204
config["reward_shift"] = 0.0
22012205

0 commit comments

Comments
 (0)