diff --git a/godot_rl/wrappers/onnx/stable_baselines_export.py b/godot_rl/wrappers/onnx/stable_baselines_export.py index 398e4f74..72c36863 100644 --- a/godot_rl/wrappers/onnx/stable_baselines_export.py +++ b/godot_rl/wrappers/onnx/stable_baselines_export.py @@ -52,6 +52,9 @@ def export_model_as_onnx(model, onnx_model_path: str, use_obs_array: bool = Fals dummy_input = None onnxable_model = None + if isinstance(model, SAC): + assert use_obs_array, "SAC ONNX export works with use_obs_array=True, MLPPolicy and SBGSingleObsEnv only." + if isinstance(model, PPO): onnxable_model = OnnxablePolicy( ["obs"],