From 5cabe478913115b74bc06c9f4376317bbcabe237 Mon Sep 17 00:00:00 2001
From: Travis <36664554+Travh98@users.noreply.github.com>
Date: Wed, 4 Dec 2024 18:49:28 -0500
Subject: [PATCH] Fixed missing import of spaces from Gymnasium

---
 godot_rl/wrappers/onnx/stable_baselines_export.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/godot_rl/wrappers/onnx/stable_baselines_export.py b/godot_rl/wrappers/onnx/stable_baselines_export.py
index 348f8c4e..1752842f 100644
--- a/godot_rl/wrappers/onnx/stable_baselines_export.py
+++ b/godot_rl/wrappers/onnx/stable_baselines_export.py
@@ -1,5 +1,5 @@
 import torch
-from gymnasium.vector.utils import spaces
+from gymnasium import spaces
 from stable_baselines3 import PPO, SAC