From 947cb6e41fb7ae28a8a622e4848cf6c6ccf685a1 Mon Sep 17 00:00:00 2001 From: Ivan-267 <61947090+Ivan-267@users.noreply.github.com> Date: Thu, 22 Aug 2024 21:26:26 +0200 Subject: [PATCH] Update test_sb3_onnx_export.py Updates onnx export method name in method call as well --- tests/test_sb3_onnx_export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_sb3_onnx_export.py b/tests/test_sb3_onnx_export.py index 362a77ed..c05845fa 100644 --- 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")