aviary.core.EnvStateMessage (#95)

Future-House · Oct 27, 2024 · a3d1068 · a3d1068
1 parent 71e02db
commit a3d1068
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/aviary/core.py b/src/aviary/core.py
@@ -9,7 +9,7 @@
     TaskDataset,
 )
 from aviary.env_client import EnvironmentClient
-from aviary.message import MalformedMessageError, Message, join
+from aviary.message import EnvStateMessage, MalformedMessageError, Message, join
 from aviary.render import Renderer
 from aviary.tools import (
     INVALID_TOOL_NAME,
@@ -38,6 +38,7 @@
     "DummyEnv",
     "DummyEnvState",
     "DummyTaskDataset",
+    "EnvStateMessage",
     "Environment",
     "EnvironmentClient",
     "Frame",