From f8459b994644f661f67b44cd302a98f8bc661c3a Mon Sep 17 00:00:00 2001
From: Thibault LSDC <78021491+ThibaultLSDC@users.noreply.github.com>
Date: Thu, 3 Oct 2024 09:16:48 -0400
Subject: [PATCH] adding AgentInfo to __init__ for convenience (#166)

---
 browsergym/experiments/src/browsergym/experiments/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/browsergym/experiments/src/browsergym/experiments/__init__.py b/browsergym/experiments/src/browsergym/experiments/__init__.py
index cf6e146e..815391f1 100644
--- a/browsergym/experiments/src/browsergym/experiments/__init__.py
+++ b/browsergym/experiments/src/browsergym/experiments/__init__.py
@@ -1,2 +1,2 @@
-from .agent import Agent
-from .loop import EnvArgs, AbstractAgentArgs, ExpArgs, get_exp_result
+from .agent import Agent, AgentInfo
+from .loop import AbstractAgentArgs, EnvArgs, ExpArgs, get_exp_result