fix loading vision policy bug

Author: ooctipus

scripts/reinforcement_learning/rsl_rl/play.py

@@ -169,7 +169,7 @@ def main(env_cfg: ManagerBasedRLEnvCfg | DirectRLEnvCfg, agent_cfg: RslRlBaseRun
     # export policy to onnx/jit
     export_model_dir = os.path.join(os.path.dirname(resume_path), "exported")
     export_policy_as_jit(policy_nn, normalizer=normalizer, path=export_model_dir, filename="policy.pt")
-    export_policy_as_onnx(policy_nn, normalizer=normalizer, path=export_model_dir, filename="policy.onnx")
+    # export_policy_as_onnx(policy_nn, normalizer=normalizer, path=export_model_dir, filename="policy.onnx")
 
     dt = env.unwrapped.step_dt
 

source/isaaclab_tasks/isaaclab_tasks/manager_based/manipulation/dexsuite/config/kuka_allegro/dexsuite_kuka_allegro_vision_env_cfg.py

@@ -197,12 +197,12 @@ class DexsuiteKukaAllegroLiftSingleCameraEnvCfg(
     pass
 
 
-# @configclass
-# class DexsuiteKukaAllegroLiftSingleCameraEnvCfg_PLAY(
-#     KukaAllegroSingleCameraMixinCfg,
-#     dexsuite_state_impl.DexsuiteLiftEnvCfg_PLAY
-# ):
-#     pass
+@configclass
+class DexsuiteKukaAllegroLiftSingleCameraEnvCfg_PLAY(
+    KukaAllegroSingleCameraMixinCfg,
+    dexsuite_state_impl.DexsuiteLiftEnvCfg_PLAY
+):
+    pass
 
 
 # DuoCamera