Merge pull request #21 from d-krupke/development

FIX: Don't call setup on listfuncs

README.rst

@@ -359,6 +359,7 @@ The project is reasonably easy:
 Changes
 -------
 
+- 0.10.1: FIX: Listing functions will no longer execute setup functions.
 -  0.10.0: `Batch` is now named `JobBundling`. There is a method `join` for easier synchronization. `exec` allows to executed commands just like `srun` and `sbatch`, but uniform syntax with other slurmified functions. Functions can now also be called with `distribute_and_wait`. If you call `python3 -m slurminade.check --partition YOUR_PARTITION --constraint YOUR_CONSTRAINT` you can check if your slurm configuration is running correctly.
 -  0.9.0: Lots of improvements.
 -  0.8.1: Bugfix and automatic detection of wrong usage when using ``Batch`` with ``wait_for``.

pyproject.toml

@@ -7,7 +7,7 @@ where = ["src"]
 
 [project]
 name = "slurminade"
-version = "0.10.0"
+version = "0.10.1"
 authors = [
     { name = "TU Braunschweig, IBR, Algorithms Group (Dominik Krupke)", email = "krupke@ibr.cs.tu-bs.de" },
 ]

src/slurminade/execute.py

@@ -12,7 +12,7 @@
 from .function import SlurmFunction
 from .function_map import FunctionMap, set_entry_point
 from .guard import prevent_distribution
-
+from .node_setup import disable_setup
 
 @click.command()
 @click.option(
@@ -37,7 +37,8 @@
 )
 def main(root, calls, fromfile, listfuncs):
     prevent_distribution()  # make sure, the code on the node does not distribute itself.
-
+    if listfuncs:
+        disable_setup()
     set_entry_point(root)
     with open(root) as f:
         code = "".join(f.readlines())

src/slurminade/node_setup.py

@@ -3,12 +3,20 @@
 
 from .guard import on_slurm_node
 
+_no_setup = False
+
+def disable_setup():
+    """
+    Disable the setup function. This is useful for testing.
+    """
+    global _no_setup
+    _no_setup = True
 
 def node_setup(func: typing.Callable):
     """
     Decorator: Call this function on the node before running any function calls.
     """
-    if on_slurm_node():
+    if on_slurm_node() and not _no_setup:
         func()
     else:
         # check if the function has no arguments