From 4355697e216ede5263081ee4cc1732935dd9f3cc Mon Sep 17 00:00:00 2001
From: Kallinteris Andreas
 <30759571+Kallinteris-Andreas@users.noreply.github.com>
Date: Sat, 17 Feb 2024 04:52:30 +0000
Subject: [PATCH] added `MazeEnv.model` and `MazeEnv.data`

---
 gymnasium_robotics/envs/maze/ant_maze.py    | 8 ++++++++
 gymnasium_robotics/envs/maze/ant_maze_v4.py | 8 ++++++++
 gymnasium_robotics/envs/maze/point_maze.py  | 8 ++++++++
 3 files changed, 24 insertions(+)

diff --git a/gymnasium_robotics/envs/maze/ant_maze.py b/gymnasium_robotics/envs/maze/ant_maze.py
index b586e9e8..d1395e78 100644
--- a/gymnasium_robotics/envs/maze/ant_maze.py
+++ b/gymnasium_robotics/envs/maze/ant_maze.py
@@ -122,3 +122,11 @@ def render(self):
     def close(self):
         super().close()
         self.ant_env.close()
+
+    @property
+    def model(self):
+        return self.ant_env.model
+
+    @property
+    def data(self):
+        return self.ant_env.data
diff --git a/gymnasium_robotics/envs/maze/ant_maze_v4.py b/gymnasium_robotics/envs/maze/ant_maze_v4.py
index da448e96..edcf7f5e 100644
--- a/gymnasium_robotics/envs/maze/ant_maze_v4.py
+++ b/gymnasium_robotics/envs/maze/ant_maze_v4.py
@@ -328,3 +328,11 @@ def render(self):
     def close(self):
         super().close()
         self.ant_env.close()
+
+    @property
+    def model(self):
+        return self.ant_env.model
+
+    @property
+    def data(self):
+        return self.ant_env.data
diff --git a/gymnasium_robotics/envs/maze/point_maze.py b/gymnasium_robotics/envs/maze/point_maze.py
index f1e5710f..e655f628 100644
--- a/gymnasium_robotics/envs/maze/point_maze.py
+++ b/gymnasium_robotics/envs/maze/point_maze.py
@@ -424,3 +424,11 @@ def render(self):
     def close(self):
         super().close()
         self.point_env.close()
+
+    @property
+    def model(self):
+        return self.point_env.model
+
+    @property
+    def data(self):
+        return self.point_env.data