FIX: bad import

d-krupke · Feb 25, 2024 · 88b554e · 88b554e
1 parent af19f32
commit 88b554e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/src/slurminade/dispatcher.py b/src/slurminade/dispatcher.py
@@ -465,13 +465,4 @@ def join():
     get_dispatcher().join()
 
 
-from .function import slurmify
 
-
-@slurmify()
-def exec(cmd: typing.Union[str, typing.List[str]]):
-    """
-    Execute a command.
-    :param cmd: The command to be executed.
-    """
-    subprocess.run(cmd, check=True)
diff --git a/src/slurminade/function.py b/src/slurminade/function.py
@@ -1,6 +1,7 @@
 import inspect
 import typing
 from enum import Enum
+import subprocess
 
 from .dispatcher import FunctionCall, dispatch, get_dispatcher
 from .function_map import FunctionMap
@@ -189,3 +190,12 @@ def dec(func) -> SlurmFunction:
         return SlurmFunction(args, func, func_id)
 
     return dec
+
+
+@slurmify()
+def exec(cmd: typing.Union[str, typing.List[str]]):
+    """
+    Execute a command.
+    :param cmd: The command to be executed.
+    """
+    subprocess.run(cmd, check=True)