diff --git a/mo_gymnasium/envs/fruit_tree/fruit_tree.py b/mo_gymnasium/envs/fruit_tree/fruit_tree.py index 81c051e5..893f16df 100644 --- a/mo_gymnasium/envs/fruit_tree/fruit_tree.py +++ b/mo_gymnasium/envs/fruit_tree/fruit_tree.py @@ -399,7 +399,6 @@ def render(self): for ind, node in enumerate(self.tree): row, index_in_row = self.ind_to_state(ind) - if (row, index_in_row) == tuple(self.current_state): img = self.agent_img font_color = (255, 0, 0) # Red digits for agent node