From 9bf6d395fff3e00ec479dc6662c3ad5d5dd188a7 Mon Sep 17 00:00:00 2001
From: Tim Toebrock <35797763+titoeb@users.noreply.github.com>
Date: Sat, 19 Dec 2020 18:31:08 +0100
Subject: [PATCH] Update mab.py

Use `env.steps` instead of `steps` which is `env.state`.
---
 kaggle_environments/envs/mab/mab.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kaggle_environments/envs/mab/mab.py b/kaggle_environments/envs/mab/mab.py
index 56c952e7..c468119e 100644
--- a/kaggle_environments/envs/mab/mab.py
+++ b/kaggle_environments/envs/mab/mab.py
@@ -127,8 +127,8 @@ def renderer(steps, env):
     board = ""
 
     for i in range(1, rounds_played):
-        actions = [agent.action for agent in steps[i]]
-        rewards = [agent.reward for agent in steps[i]]
+        actions = [agent.action for agent in env.steps[i]]
+        rewards = [agent.reward for agent in env.steps[i]]
         board += f"Round {i} Actions: {actions}, Rewards: {rewards}\n"
 
     return board