Skip to content

Commit fa71023

Browse files
authored
Remove Orchestrator (#638)
## Summary Remove Orchestrator ## Detailed description - Orchestrator was added in `v0.1.0` to temporarily give a customer a shim to transition. Removing for v0.3.
1 parent 7f42d0d commit fa71023

7 files changed

Lines changed: 1 addition & 65 deletions

File tree

isaaclab_arena/embodiments/embodiment_base.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010

1111
from isaaclab_arena.assets.asset import Asset
1212
from isaaclab_arena.embodiments.common.arm_mode import ArmMode
13-
from isaaclab_arena.environments.isaaclab_arena_manager_based_env import IsaacLabArenaManagerBasedRLEnvCfg
1413
from isaaclab_arena.utils.cameras import make_camera_observation_cfg
1514
from isaaclab_arena.utils.configclass import combine_configclass_instances
1615
from isaaclab_arena.utils.pose import Pose
@@ -112,9 +111,6 @@ def get_recorder_term_cfg(self) -> RecorderManagerBaseCfg:
112111
def get_termination_cfg(self) -> Any:
113112
return self.termination_cfg
114113

115-
def modify_env_cfg(self, env_cfg: IsaacLabArenaManagerBasedRLEnvCfg) -> IsaacLabArenaManagerBasedRLEnvCfg:
116-
return env_cfg
117-
118114
def get_embodiment_name_in_scene(self) -> str:
119115
return "robot"
120116

isaaclab_arena/environments/arena_env_builder.py

Lines changed: 1 addition & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,6 @@ def __init__(self, arena_env: IsaacLabArenaEnvironment, args: argparse.Namespace
5151
)
5252
self._placement_event_cfg: EventTermCfg | None = None
5353

54-
def orchestrate(self) -> None:
55-
"""Orchestrate the environment member interaction"""
56-
if self.arena_env.orchestrator is not None:
57-
self.arena_env.orchestrator.orchestrate(
58-
self.arena_env.embodiment, self.arena_env.scene, self.arena_env.task
59-
)
60-
6154
def _solve_relations(self) -> None:
6255
"""Solve spatial relations for objects in the scene.
6356
@@ -181,19 +174,6 @@ def _modify_recorder_cfg_dataset_filename(self, recorder_cfg: RecorderManagerBas
181174
)
182175
return recorder_cfg
183176

184-
# This method gives the arena environment a chance to modify the environment configuration.
185-
# This is a workaround to allow user to gradually move to the new configuration system.
186-
# THE ORDER MATTERS HERE.
187-
# THIS WILL BE REMOVED IN THE FUTURE.
188-
def modify_env_cfg(self, env_cfg: IsaacLabArenaManagerBasedRLEnvCfg) -> IsaacLabArenaManagerBasedRLEnvCfg:
189-
"""Modify the environment configuration."""
190-
if self.arena_env.task is not None:
191-
env_cfg = self.arena_env.task.modify_env_cfg(env_cfg)
192-
if self.arena_env.embodiment is not None:
193-
env_cfg = self.arena_env.embodiment.modify_env_cfg(env_cfg)
194-
env_cfg = self.arena_env.scene.modify_env_cfg(env_cfg)
195-
return env_cfg
196-
197177
def compose_manager_cfg(self) -> IsaacLabArenaManagerBasedRLEnvCfg:
198178
"""Return base ManagerBased cfg (scene+events+terminations+xr), no registration."""
199179

@@ -351,7 +331,7 @@ def compose_manager_cfg(self) -> IsaacLabArenaManagerBasedRLEnvCfg:
351331
env_cfg.sim.physics = getattr(ArenaPhysicsCfg(), presets)
352332

353333
# Set replicate_physics for shared physics representations.
354-
# For Newton, wihotut this flag, the simulation initialization
334+
# For Newton, without this flag, the simulation initialization
355335
# takes a very long time for large number of parallel environments.
356336
if presets == "newton":
357337
env_cfg.scene.replicate_physics = True
@@ -374,12 +354,7 @@ def build_registered(
374354
) -> tuple[str, IsaacLabArenaManagerBasedRLEnvCfg]:
375355
"""Register Gym env and parse runtime cfg."""
376356
name = self.arena_env.name
377-
# orchestrate the environment member interaction
378-
self.orchestrate()
379357
cfg_entry = env_cfg if env_cfg is not None else self.compose_manager_cfg()
380-
# THIS IS A WORKAROUND TO ALLOW USER TO GRADUALLY MOVE TO THE NEW CONFIGURATION SYSTEM.
381-
# THIS WILL BE REMOVED IN THE FUTURE.
382-
cfg_entry = self.modify_env_cfg(cfg_entry)
383358
entry_point = self.get_entry_point()
384359
# Register the environment with the Gym registry.
385360
kwargs = {

isaaclab_arena/environments/isaaclab_arena_environment.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
from isaaclab_arena.assets.teleop_device_base import TeleopDeviceBase
1313
from isaaclab_arena.embodiments.embodiment_base import EmbodimentBase
1414
from isaaclab_arena.environments.isaaclab_arena_manager_based_env import IsaacLabArenaManagerBasedRLEnvCfg
15-
from isaaclab_arena.orchestrator.orchestrator_base import OrchestratorBase
1615
from isaaclab_arena.scene.scene import Scene
1716
from isaaclab_arena.tasks.task_base import TaskBase
1817

@@ -27,7 +26,6 @@ def __init__(
2726
embodiment: EmbodimentBase | None = None,
2827
task: TaskBase | None = None,
2928
teleop_device: TeleopDeviceBase | None = None,
30-
orchestrator: OrchestratorBase | None = None,
3129
env_cfg_callback: Callable[IsaacLabArenaManagerBasedRLEnvCfg] | None = None,
3230
rl_framework_entry_point: str | None = None,
3331
rl_policy_cfg: str | None = None,
@@ -39,7 +37,6 @@ def __init__(
3937
embodiment: The embodiment to use in the environment.
4038
task: The task to use in the environment.
4139
teleop_device: The teleop device to use in the environment.
42-
orchestrator: The orchestrator to use in the environment.
4340
env_cfg_callback: A callback function that modifies the environment configuration.
4441
rl_framework_entry_point: Gym kwargs key under which the RL policy config is
4542
registered. This is an IsaacLab convention: each supported RL framework has a
@@ -55,7 +52,6 @@ def __init__(
5552
self.embodiment = embodiment
5653
self.task = task
5754
self.teleop_device = teleop_device
58-
self.orchestrator = orchestrator
5955
self.env_cfg_callback = env_cfg_callback
6056
if (rl_framework_entry_point is None) != (rl_policy_cfg is None):
6157
raise ValueError("rl_framework_entry_point and rl_policy_cfg must both be set or both be None.")

isaaclab_arena/orchestrator/__init__.py

Lines changed: 0 additions & 4 deletions
This file was deleted.

isaaclab_arena/orchestrator/orchestrator_base.py

Lines changed: 0 additions & 19 deletions
This file was deleted.

isaaclab_arena/scene/scene.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
from isaaclab_arena.assets.object_base import ObjectType
1717
from isaaclab_arena.assets.object_reference import ObjectReference
1818
from isaaclab_arena.assets.object_set import RigidObjectSet
19-
from isaaclab_arena.environments.isaaclab_arena_manager_based_env import IsaacLabArenaManagerBasedRLEnvCfg
2019
from isaaclab_arena.utils.configclass import make_configclass
2120
from isaaclab_arena.utils.phyx_utils import add_contact_report
2221

@@ -112,9 +111,6 @@ def get_objects_with_relations(self) -> list[Object | ObjectReference]:
112111
objects_with_relations.append(asset)
113112
return objects_with_relations
114113

115-
def modify_env_cfg(self, env_cfg: IsaacLabArenaManagerBasedRLEnvCfg) -> IsaacLabArenaManagerBasedRLEnvCfg:
116-
return env_cfg
117-
118114
def export_to_usd(self, output_path: pathlib.Path, root_prim_path: str = "/World") -> None:
119115
"""Exports the scene to a USD file.
120116

isaaclab_arena/tasks/task_base.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from isaaclab.managers.recorder_manager import RecorderManagerBaseCfg
1111

1212
from isaaclab_arena.embodiments.common.arm_mode import ArmMode
13-
from isaaclab_arena.environments.isaaclab_arena_manager_based_env import IsaacLabArenaManagerBasedRLEnvCfg
1413
from isaaclab_arena.metrics.metric_base import MetricBase
1514

1615

@@ -55,9 +54,6 @@ def get_commands_cfg(self) -> Any:
5554
def get_recorder_term_cfg(self) -> RecorderManagerBaseCfg:
5655
return None
5756

58-
def modify_env_cfg(self, env_cfg: IsaacLabArenaManagerBasedRLEnvCfg) -> IsaacLabArenaManagerBasedRLEnvCfg:
59-
return env_cfg
60-
6157
def get_viewer_cfg(self) -> ViewerCfg:
6258
return ViewerCfg()
6359

0 commit comments

Comments
 (0)