fix: ValueError: too many values to unpack (expected 4)

This commit is contained in:
Krzysztof Rudnicki 2023-05-21 19:18:50 +02:00
parent a605caef2b
commit b8522b90bb

View File

@ -131,7 +131,7 @@ def inference(env, q_table):
# Choose the action with the highest Q-value
action = np.argmax(q_table)
# Take the action and observe the next state
_, terminated, truncated, _ = env.step(action)
_, _, terminated, truncated, _ = env.step(action)
done = terminated or truncated