Remove proxyable command

Author: Al Rigazzi

smartsim/_core/launcher/step/alpsStep.py

@@ -32,7 +32,7 @@
 from ....error import AllocationError
 from ....log import get_logger
 from ....settings import AprunSettings, RunSettings, Singularity
-from .step import Step, proxyable_launch_cmd
+from .step import Step
 
 logger = get_logger(__name__)
 
@@ -57,7 +57,6 @@ def _get_mpmd(self) -> t.List[RunSettings]:
         """
         return self.run_settings.mpmd
 
-    @proxyable_launch_cmd
     def get_launch_cmd(self) -> t.List[str]:
         """Get the command to launch this step
 

smartsim/_core/launcher/step/localStep.py

@@ -30,7 +30,7 @@
 
 from ....settings import Singularity
 from ....settings.base import RunSettings
-from .step import Step, proxyable_launch_cmd
+from .step import Step
 
 
 class LocalStep(Step):
@@ -43,7 +43,6 @@ def __init__(self, name: str, cwd: str, run_settings: RunSettings):
     def env(self) -> t.Dict[str, str]:
         return self._env
 
-    @proxyable_launch_cmd
     def get_launch_cmd(self) -> t.List[str]:
         cmd = []
 

smartsim/_core/launcher/step/mpiStep.py

@@ -33,7 +33,7 @@
 from ....log import get_logger
 from ....settings import MpiexecSettings, MpirunSettings, OrterunSettings
 from ....settings.base import RunSettings
-from .step import Step, proxyable_launch_cmd
+from .step import Step
 
 logger = get_logger(__name__)
 
@@ -56,7 +56,6 @@ def __init__(self, name: str, cwd: str, run_settings: RunSettings) -> None:
 
     _supported_launchers = ["PBS", "SLURM", "LSB", "SGE"]
 
-    @proxyable_launch_cmd
     def get_launch_cmd(self) -> t.List[str]:
         """Get the command to launch this step
 

smartsim/_core/launcher/step/step.py

@@ -27,7 +27,6 @@
 from __future__ import annotations
 
 import copy
-import functools
 import os.path as osp
 import pathlib
 import time
@@ -127,24 +126,3 @@ def add_to_batch(self, step: Step) -> None:
         :param step: a job step instance e.g. SrunStep
         """
         raise SmartSimError("add_to_batch not implemented for this step type")
-
-
-_StepT = t.TypeVar("_StepT", bound=Step)
-
-
-def proxyable_launch_cmd(
-    fn: t.Callable[[_StepT], t.List[str]], /
-) -> t.Callable[[_StepT], t.List[str]]:
-    @functools.wraps(fn)
-    def _get_launch_cmd(self: _StepT) -> t.List[str]:
-        """
-        Generate a launch command that executes the `JobStep` directly.
-
-        Steps implementing `get_launch_cmd` and decorated with
-        `proxyable_launch_cmd` support direct launching."""
-        original_cmd_list = fn(self)
-
-        # Always use direct launch
-        return original_cmd_list
-
-    return _get_launch_cmd