Unverified Commit aa5472c6 authored by David Hoeller's avatar David Hoeller Committed by GitHub

Fixes running train scripts when rsl_rl is not installed (#784)

# Description

Previously, it was not possible to run training scripts with rl_games,
skrl, sb3 when rsl_rl was not installed.

Fixes #749 

## Type of change

- Bug fix (non-breaking change which fixes an issue)

## Checklist

- [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with
`./isaaclab.sh --format`
- [ ] I have made corresponding changes to the documentation
- [x] My changes generate no new warnings
- [ ] I have added tests that prove my fix is effective or that my
feature works
- [ ] I have updated the changelog and the corresponding version in the
extension's `config/extension.toml` file
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there
parent f75a51f3
...@@ -96,7 +96,6 @@ class DirectRLEnv(gym.Env): ...@@ -96,7 +96,6 @@ class DirectRLEnv(gym.Env):
print(f"\tPhysics step-size : {self.physics_dt}") print(f"\tPhysics step-size : {self.physics_dt}")
print(f"\tRendering step-size : {self.physics_dt * self.cfg.sim.render_interval}") print(f"\tRendering step-size : {self.physics_dt * self.cfg.sim.render_interval}")
print(f"\tEnvironment step-size : {self.step_dt}") print(f"\tEnvironment step-size : {self.step_dt}")
print(f"\tEnvironment device: {self.device}")
if self.cfg.sim.render_interval < self.cfg.decimation: if self.cfg.sim.render_interval < self.cfg.decimation:
msg = ( msg = (
......
...@@ -23,6 +23,6 @@ gym.register( ...@@ -23,6 +23,6 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": AllegroHandEnvCfg, "env_cfg_entry_point": AllegroHandEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.AllegroHandPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AllegroHandPPORunnerCfg",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": AntEnvCfg, "env_cfg_entry_point": AntEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.AntPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}.rsl_rl_ppo_cfg:AntPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": AnymalCFlatEnvCfg, "env_cfg_entry_point": AnymalCFlatEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.AnymalCFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -35,7 +35,7 @@ gym.register( ...@@ -35,7 +35,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": AnymalCRoughEnvCfg, "env_cfg_entry_point": AnymalCRoughEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.AnymalCRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -24,7 +24,7 @@ gym.register( ...@@ -24,7 +24,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": CartpoleEnvCfg, "env_cfg_entry_point": CartpoleEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.CartpolePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CartpolePPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml", "sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml",
}, },
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg # noqa: F401, F403
...@@ -22,7 +22,7 @@ gym.register( ...@@ -22,7 +22,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": FrankaCabinetEnvCfg, "env_cfg_entry_point": FrankaCabinetEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.FrankaCabinetPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.FrankaCabinetPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": HumanoidEnvCfg, "env_cfg_entry_point": HumanoidEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.HumanoidPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.HumanoidPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": QuadcopterEnvCfg, "env_cfg_entry_point": QuadcopterEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.QuadcopterPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.QuadcopterPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": ShadowHandEnvCfg, "env_cfg_entry_point": ShadowHandEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.ShadowHandPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.ShadowHandPPORunnerCfg",
}, },
) )
...@@ -34,7 +34,7 @@ gym.register( ...@@ -34,7 +34,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": ShadowHandOpenAIEnvCfg, "env_cfg_entry_point": ShadowHandOpenAIEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_ff_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_ff_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.ShadowHandAsymFFPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.ShadowHandAsymFFPPORunnerCfg",
}, },
) )
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -21,7 +21,7 @@ gym.register( ...@@ -21,7 +21,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": ant_env_cfg.AntEnvCfg, "env_cfg_entry_point": ant_env_cfg.AntEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.AntPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AntPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml", "sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml",
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg
...@@ -23,7 +23,7 @@ gym.register( ...@@ -23,7 +23,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": CartpoleEnvCfg, "env_cfg_entry_point": CartpoleEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.CartpolePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CartpolePPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml", "sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml",
}, },
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg # noqa: F401, F403
...@@ -21,7 +21,7 @@ gym.register( ...@@ -21,7 +21,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": humanoid_env_cfg.HumanoidEnvCfg, "env_cfg_entry_point": humanoid_env_cfg.HumanoidEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_ppo_cfg.HumanoidPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.HumanoidPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml", "sb3_cfg_entry_point": f"{agents.__name__}:sb3_ppo_cfg.yaml",
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeA1FlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.UnitreeA1FlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeA1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeA1FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -28,7 +28,7 @@ gym.register( ...@@ -28,7 +28,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeA1FlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.UnitreeA1FlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeA1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeA1FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -39,7 +39,7 @@ gym.register( ...@@ -39,7 +39,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeA1RoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.UnitreeA1RoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeA1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeA1RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -50,7 +50,7 @@ gym.register( ...@@ -50,7 +50,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeA1RoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.UnitreeA1RoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeA1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeA1RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -16,7 +16,7 @@ gym.register( ...@@ -16,7 +16,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalBFlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.AnymalBFlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalBFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalBFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -27,7 +27,7 @@ gym.register( ...@@ -27,7 +27,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalBFlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.AnymalBFlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalBFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalBFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -38,7 +38,7 @@ gym.register( ...@@ -38,7 +38,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalBRoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.AnymalBRoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalBRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalBRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -49,7 +49,7 @@ gym.register( ...@@ -49,7 +49,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalBRoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.AnymalBRoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalBRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalBRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalCFlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.AnymalCFlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalCFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCFlatPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
...@@ -30,7 +30,7 @@ gym.register( ...@@ -30,7 +30,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalCFlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.AnymalCFlatEnvCfg_PLAY,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_flat_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalCFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -42,7 +42,7 @@ gym.register( ...@@ -42,7 +42,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalCRoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.AnymalCRoughEnvCfg,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalCRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -54,7 +54,7 @@ gym.register( ...@@ -54,7 +54,7 @@ gym.register(
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalCRoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.AnymalCRoughEnvCfg_PLAY,
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_rough_ppo_cfg.yaml",
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalCRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalCRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalDFlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.AnymalDFlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalDFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalDFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -28,7 +28,7 @@ gym.register( ...@@ -28,7 +28,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.AnymalDFlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.AnymalDFlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalDFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalDFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -39,7 +39,7 @@ gym.register( ...@@ -39,7 +39,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalDRoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.AnymalDRoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalDRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalDRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -50,7 +50,7 @@ gym.register( ...@@ -50,7 +50,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.AnymalDRoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.AnymalDRoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AnymalDRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AnymalDRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.CassieFlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.CassieFlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CassieFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CassieFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -28,7 +28,7 @@ gym.register( ...@@ -28,7 +28,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.CassieFlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.CassieFlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CassieFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CassieFlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -39,7 +39,7 @@ gym.register( ...@@ -39,7 +39,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.CassieRoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.CassieRoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CassieRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CassieRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -50,7 +50,7 @@ gym.register( ...@@ -50,7 +50,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.CassieRoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.CassieRoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CassieRoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CassieRoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -18,7 +18,7 @@ gym.register( ...@@ -18,7 +18,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.G1RoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.G1RoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.G1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.G1RoughPPORunnerCfg",
}, },
) )
...@@ -29,7 +29,7 @@ gym.register( ...@@ -29,7 +29,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.G1RoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.G1RoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.G1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.G1RoughPPORunnerCfg",
}, },
) )
...@@ -40,7 +40,7 @@ gym.register( ...@@ -40,7 +40,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.G1FlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.G1FlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.G1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.G1FlatPPORunnerCfg",
}, },
) )
...@@ -51,6 +51,6 @@ gym.register( ...@@ -51,6 +51,6 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.G1FlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.G1FlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.G1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.G1FlatPPORunnerCfg",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeGo1FlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.UnitreeGo1FlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo1FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -28,7 +28,7 @@ gym.register( ...@@ -28,7 +28,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeGo1FlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.UnitreeGo1FlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo1FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -39,7 +39,7 @@ gym.register( ...@@ -39,7 +39,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeGo1RoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.UnitreeGo1RoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo1RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -50,7 +50,7 @@ gym.register( ...@@ -50,7 +50,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeGo1RoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.UnitreeGo1RoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo1RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeGo2FlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.UnitreeGo2FlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo2FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo2FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -28,7 +28,7 @@ gym.register( ...@@ -28,7 +28,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.UnitreeGo2FlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.UnitreeGo2FlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo2FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo2FlatPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_flat_ppo_cfg.yaml",
}, },
) )
...@@ -39,7 +39,7 @@ gym.register( ...@@ -39,7 +39,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeGo2RoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.UnitreeGo2RoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo2RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo2RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -50,7 +50,7 @@ gym.register( ...@@ -50,7 +50,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.UnitreeGo2RoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.UnitreeGo2RoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.UnitreeGo2RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.UnitreeGo2RoughPPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_rough_ppo_cfg.yaml",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -18,7 +18,7 @@ gym.register( ...@@ -18,7 +18,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.H1RoughEnvCfg, "env_cfg_entry_point": rough_env_cfg.H1RoughEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.H1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.H1RoughPPORunnerCfg",
}, },
) )
...@@ -29,7 +29,7 @@ gym.register( ...@@ -29,7 +29,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": rough_env_cfg.H1RoughEnvCfg_PLAY, "env_cfg_entry_point": rough_env_cfg.H1RoughEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.H1RoughPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.H1RoughPPORunnerCfg",
}, },
) )
...@@ -40,7 +40,7 @@ gym.register( ...@@ -40,7 +40,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.H1FlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.H1FlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.H1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.H1FlatPPORunnerCfg",
}, },
) )
...@@ -51,6 +51,6 @@ gym.register( ...@@ -51,6 +51,6 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.H1FlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.H1FlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.H1FlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.H1FlatPPORunnerCfg",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.SpotFlatEnvCfg, "env_cfg_entry_point": flat_env_cfg.SpotFlatEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.SpotFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.SpotFlatPPORunnerCfg",
}, },
) )
...@@ -27,6 +27,6 @@ gym.register( ...@@ -27,6 +27,6 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": flat_env_cfg.SpotFlatEnvCfg_PLAY, "env_cfg_entry_point": flat_env_cfg.SpotFlatEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.SpotFlatPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.SpotFlatPPORunnerCfg",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -20,7 +20,7 @@ gym.register( ...@@ -20,7 +20,7 @@ gym.register(
entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv", entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv",
kwargs={ kwargs={
"env_cfg_entry_point": joint_pos_env_cfg.FrankaCabinetEnvCfg, "env_cfg_entry_point": joint_pos_env_cfg.FrankaCabinetEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CabinetPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CabinetPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
...@@ -32,7 +32,7 @@ gym.register( ...@@ -32,7 +32,7 @@ gym.register(
entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv", entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv",
kwargs={ kwargs={
"env_cfg_entry_point": joint_pos_env_cfg.FrankaCabinetEnvCfg_PLAY, "env_cfg_entry_point": joint_pos_env_cfg.FrankaCabinetEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.CabinetPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.CabinetPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -21,7 +21,7 @@ gym.register( ...@@ -21,7 +21,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": allegro_env_cfg.AllegroCubeEnvCfg, "env_cfg_entry_point": allegro_env_cfg.AllegroCubeEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AllegroCubePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AllegroCubePPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
...@@ -33,7 +33,7 @@ gym.register( ...@@ -33,7 +33,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": allegro_env_cfg.AllegroCubeEnvCfg_PLAY, "env_cfg_entry_point": allegro_env_cfg.AllegroCubeEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AllegroCubePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AllegroCubePPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
...@@ -49,7 +49,7 @@ gym.register( ...@@ -49,7 +49,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": allegro_env_cfg.AllegroCubeNoVelObsEnvCfg, "env_cfg_entry_point": allegro_env_cfg.AllegroCubeNoVelObsEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AllegroCubeNoVelObsPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AllegroCubeNoVelObsPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
...@@ -61,7 +61,7 @@ gym.register( ...@@ -61,7 +61,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": allegro_env_cfg.AllegroCubeNoVelObsEnvCfg_PLAY, "env_cfg_entry_point": allegro_env_cfg.AllegroCubeNoVelObsEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.AllegroCubeNoVelObsPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.AllegroCubeNoVelObsPPORunnerCfg",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
}, },
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -20,7 +20,7 @@ gym.register( ...@@ -20,7 +20,7 @@ gym.register(
entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv", entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv",
kwargs={ kwargs={
"env_cfg_entry_point": joint_pos_env_cfg.FrankaCubeLiftEnvCfg, "env_cfg_entry_point": joint_pos_env_cfg.FrankaCubeLiftEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.LiftCubePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.LiftCubePPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
}, },
...@@ -32,7 +32,7 @@ gym.register( ...@@ -32,7 +32,7 @@ gym.register(
entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv", entry_point="omni.isaac.lab.envs:ManagerBasedRLEnv",
kwargs={ kwargs={
"env_cfg_entry_point": joint_pos_env_cfg.FrankaCubeLiftEnvCfg_PLAY, "env_cfg_entry_point": joint_pos_env_cfg.FrankaCubeLiftEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.LiftCubePPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.LiftCubePPORunnerCfg",
"skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml", "skrl_cfg_entry_point": f"{agents.__name__}:skrl_ppo_cfg.yaml",
"rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml", "rl_games_cfg_entry_point": f"{agents.__name__}:rl_games_ppo_cfg.yaml",
}, },
......
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_ppo_cfg # noqa: F401, F403
...@@ -17,7 +17,7 @@ gym.register( ...@@ -17,7 +17,7 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": navigation_env_cfg.NavigationEnvCfg, "env_cfg_entry_point": navigation_env_cfg.NavigationEnvCfg,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.NavigationEnvPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.NavigationEnvPPORunnerCfg",
}, },
) )
...@@ -27,6 +27,6 @@ gym.register( ...@@ -27,6 +27,6 @@ gym.register(
disable_env_checker=True, disable_env_checker=True,
kwargs={ kwargs={
"env_cfg_entry_point": navigation_env_cfg.NavigationEnvCfg_PLAY, "env_cfg_entry_point": navigation_env_cfg.NavigationEnvCfg_PLAY,
"rsl_rl_cfg_entry_point": agents.rsl_rl_cfg.NavigationEnvPPORunnerCfg, "rsl_rl_cfg_entry_point": f"{agents.__name__}:rsl_rl_ppo_cfg.NavigationEnvPPORunnerCfg",
}, },
) )
...@@ -2,5 +2,3 @@ ...@@ -2,5 +2,3 @@
# All rights reserved. # All rights reserved.
# #
# SPDX-License-Identifier: BSD-3-Clause # SPDX-License-Identifier: BSD-3-Clause
from . import rsl_rl_cfg # noqa: F401, F403
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment