Update test_sb3_onnx_export.py

Updates onnx export method name in method call as well
edbeeching · Aug 22, 2024 · 947cb6e · 947cb6e
1 parent aa4e216
commit 947cb6e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_sb3_onnx_export.py b/tests/test_sb3_onnx_export.py
@@ -39,6 +39,6 @@ def test_pytorch_vs_onnx(env_name, port):
         tensorboard_log="logs/log",
     )
 
-    export_ppo_model_as_onnx(ppo, f"{env_name}_tmp.onnx")
+    export_model_as_onnx(ppo, f"{env_name}_tmp.onnx")
     verify_onnx_export(ppo, f"{env_name}_tmp.onnx")
     os.remove(f"{env_name}_tmp.onnx")