diff --git a/src/ansys/hps/client/jms/resource/task_definition.py b/src/ansys/hps/client/jms/resource/task_definition.py index a1b0f2a10..cc7ca8a74 100644 --- a/src/ansys/hps/client/jms/resource/task_definition.py +++ b/src/ansys/hps/client/jms/resource/task_definition.py @@ -52,6 +52,8 @@ class HpcResources(Object): Whether a job can't share resources with other running jobs. queue : str, optional Name of the job scheduler queue. + use_local_scratch : bool, optional + Whether to use node local storage. """ class Meta: @@ -64,12 +66,14 @@ def __init__( num_gpus_per_node: int = missing, exclusive: bool = missing, queue: str = missing, + use_local_scratch: bool = missing, **kwargs, ): self.num_cores_per_node = num_cores_per_node self.num_gpus_per_node = num_gpus_per_node self.exclusive = exclusive self.queue = queue + self.use_local_scratch = use_local_scratch self.obj_type = self.__class__.__name__ diff --git a/src/ansys/hps/client/jms/schema/task_definition.py b/src/ansys/hps/client/jms/schema/task_definition.py index bf1755eb2..506874e87 100644 --- a/src/ansys/hps/client/jms/schema/task_definition.py +++ b/src/ansys/hps/client/jms/schema/task_definition.py @@ -54,6 +54,10 @@ class Meta: queue = fields.Str( allow_none=True, metadata={"description": "Name of the job scheduler queue."} ) + use_local_scratch = fields.Bool( + allow_none=True, + metadata={"description": "Whether to use node local storage."}, + ) class ResourceRequirementsSchema(BaseSchema):