Update extended.py

pyiron · May 9, 2024 · 6cf81d5 · 6cf81d5
1 parent fced607
commit 6cf81d5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pylammpsmpi/wrapper/extended.py b/pylammpsmpi/wrapper/extended.py
@@ -248,6 +248,7 @@ def __init__(
         client=None,
         mode="local",
         cmdargs=None,
+        use_srun=False,
     ):
         self.cores = cores
         self.working_directory = working_directory
@@ -259,6 +260,7 @@ def __init__(
             oversubscribe=self.oversubscribe,
             working_directory=self.working_directory,
             cmdargs=cmdargs,
+            use_srun=use_srun,
         )
 
     def __getattr__(self, name):