From 6f298b4435893dceda5733e1a4a1b91882546f3b Mon Sep 17 00:00:00 2001
From: Ziluo Ding <leondg@outlook.com>
Date: Mon, 1 Jul 2024 16:26:40 +0800
Subject: [PATCH] Update helpers.py

---
 legged_gym/utils/helpers.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/legged_gym/utils/helpers.py b/legged_gym/utils/helpers.py
index dc231af..7dc9716 100644
--- a/legged_gym/utils/helpers.py
+++ b/legged_gym/utils/helpers.py
@@ -157,8 +157,7 @@ def export_policy_as_jit(actor_critic, path):
         path = os.path.join(path, 'policy_1.pt')
         model = copy.deepcopy(actor_critic.actor).to('cpu')
         traced_script_module = torch.jit.script(model)
-        traced_script_module.save(path)
-
+        traced_script_module.save(path) 
 
 class PolicyExporterLSTM(torch.nn.Module):
     def __init__(self, actor_critic):