From 6cf81d52b28144f63b0bb1c7910eaebbc45ce685 Mon Sep 17 00:00:00 2001
From: Jan Janssen <jan-janssen@users.noreply.github.com>
Date: Thu, 9 May 2024 00:23:08 -0500
Subject: [PATCH] Update extended.py

---
 pylammpsmpi/wrapper/extended.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pylammpsmpi/wrapper/extended.py b/pylammpsmpi/wrapper/extended.py
index d9d5723..29a776d 100644
--- 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):