@@ -252,12 +252,15 @@ Looking at `compute_obs()` from Step 1, the observation is computed from three s
252252(` foo ` , ` bar ` , ` baz ` ) and the previous actions:
253253
254254``` python
255- torch.cat([
256- self .data_source.foo + 1.0 ,
257- self .data_source.bar + 2.0 * self .data_source.baz,
258- self .data_source.baz,
259- self ._actions,
260- ], dim = - 1 )
255+ torch.cat(
256+ [
257+ self .data_source.foo + 1.0 ,
258+ self .data_source.bar + 2.0 * self .data_source.baz,
259+ self .data_source.baz,
260+ self ._actions,
261+ ],
262+ dim = - 1 ,
263+ )
261264```
262265
263266Each of these tensors needs to be declared so the exporter knows how to feed data into the ONNX
@@ -780,7 +783,10 @@ import torch
780783from isaaclab.assets import Articulation
781784from isaaclab.managers import SceneEntityCfg
782785
783- from exploy.exporter.frameworks.isaaclab.derived_tensors import body_link_ang_vel_b, body_link_lin_vel_b
786+ from exploy.exporter.frameworks.isaaclab.derived_tensors import (
787+ body_link_ang_vel_b,
788+ body_link_lin_vel_b,
789+ )
784790
785791
786792def body_lin_vel_b (env , asset_cfg = SceneEntityCfg(" robot" )) -> torch.Tensor:
@@ -984,7 +990,9 @@ env = Environment(data_source=data_source)
984990exp_env = ExportableEnv(env = env)
985991actor = Actor(num_obs = env.num_obs, num_act = env.num_act).eval()
986992
987- assert export_and_evaluate(exp_env, actor, " policy.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20 )
993+ assert export_and_evaluate(
994+ exp_env, actor, " policy.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20
995+ )
988996```
989997
990998### Environment with a torch module
@@ -996,7 +1004,9 @@ exp_env = ExportableEnv(env=env)
9961004actor = Actor(num_obs = env.num_obs, num_act = env.num_act).eval()
9971005exp_env.context_manager().add_module(env.module)
9981006
999- assert export_and_evaluate(exp_env, actor, " policy_with_module.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20 )
1007+ assert export_and_evaluate(
1008+ exp_env, actor, " policy_with_module.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20
1009+ )
10001010```
10011011
10021012### Environment with a torch module and an RNN actor
@@ -1014,5 +1024,7 @@ add_actor_memory(
10141024 get_hidden_states_func = actor.get_state,
10151025)
10161026
1017- assert export_and_evaluate(exp_env, actor, " policy_with_rnn.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20 )
1027+ assert export_and_evaluate(
1028+ exp_env, actor, " policy_with_rnn.onnx" , num_eval_episodes = 1 , max_eval_steps_per_episode = 20
1029+ )
10181030```
0 commit comments