Update extended.py

Author: jan-janssen

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):