Fixed missing import of spaces from Gymnasium

edbeeching · Dec 4, 2024 · 5cabe47 · 5cabe47
1 parent e1d89c9
commit 5cabe47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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