diff --git a/src/slurminade/__init__.py b/src/slurminade/__init__.py index c022fe3..784beb9 100644 --- a/src/slurminade/__init__.py +++ b/src/slurminade/__init__.py @@ -55,7 +55,7 @@ def clean_up(): """ # flake8: noqa F401 -from .function import slurmify +from .function import slurmify, shell from .conf import update_default_configuration, set_default_configuration from .guard import ( set_dispatch_limit, @@ -94,6 +94,7 @@ def clean_up(): "TestDispatcher", "SubprocessDispatcher", "set_entry_point", + "shell", "node_setup", ] diff --git a/src/slurminade/function.py b/src/slurminade/function.py index 3ec9fe2..b9d56fd 100644 --- a/src/slurminade/function.py +++ b/src/slurminade/function.py @@ -216,7 +216,7 @@ def dec(func) -> SlurmFunction: @slurmify() -def exec(cmd: typing.Union[str, typing.List[str]]): +def shell(cmd: typing.Union[str, typing.List[str]]): """ Execute a command. :param cmd: The command to be executed.