Skip to content

Commit eb5c00e

Browse files
Update to use Taxi-v4
1 parent 4436f89 commit eb5c00e

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

docs/tutorials/training_agents/action_masking_taxi.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -170,7 +170,7 @@ def train_q_learning(
170170
print(f"Run {i + 1}/{n_runs} with seed {seed}")
171171

172172
# Train agent WITH action masking
173-
env_masked = gym.make("Taxi-v3")
173+
env_masked = gym.make("Taxi-v4")
174174
masked_results = train_q_learning(
175175
env_masked,
176176
use_action_mask=True,
@@ -184,7 +184,7 @@ def train_q_learning(
184184
masked_results_list.append(masked_results)
185185

186186
# Train agent WITHOUT action masking
187-
env_unmasked = gym.make("Taxi-v3")
187+
env_unmasked = gym.make("Taxi-v4")
188188
unmasked_results = train_q_learning(
189189
env_unmasked,
190190
use_action_mask=False,

0 commit comments

Comments
 (0)