Set number of tasks for PALS mpiexec (#247)

Correction to flag setting number of tasks for PalsMpiexecSettings

[ committed by @rickybalin ]
[ reviewed by @MattToast, @ashao ]

smartsim/settings/mpirunSettings.py

@@ -35,4 +35,4 @@
     "mpirunSettings will be deprecated; use mpiSettings instead.",
     DeprecationWarning,
     stacklevel=2
-)
+)

smartsim/settings/palsSettings.py

@@ -127,6 +127,14 @@ def set_cpu_binding_type(self, bind_type):
         """
         self.run_args["cpu-bind"] = str(bind_type)
 
+    def set_tasks(self, tasks):
+        """Set the number of tasks
+
+        :param tasks: number of total tasks to launch
+        :type tasks: int
+        """
+        self.run_args["np"] = int(tasks)
+
     def set_tasks_per_node(self, tasks_per_node):
         """Set the number of tasks per node