diff --git a/v3io_gputils/__init__.py b/v3io_gputils/__init__.py index 2792152..9123cf0 100644 --- a/v3io_gputils/__init__.py +++ b/v3io_gputils/__init__.py @@ -1 +1 @@ -__version__ = '0.0.7' +__version__ = '0.0.8' diff --git a/v3io_gputils/mpijob.py b/v3io_gputils/mpijob.py index 20f839e..f74959f 100644 --- a/v3io_gputils/mpijob.py +++ b/v3io_gputils/mpijob.py @@ -21,6 +21,7 @@ 'name': '', 'command': [], 'volumeMounts': [{'name': 'v3io', 'mountPath': '/User'}], + 'workingDir': '/User', 'securityContext': { 'capabilities': {'add': ['IPC_LOCK']}}, 'resources': { @@ -111,6 +112,10 @@ def replicas(self, replicas_num): self._struct['spec']['replicas'] = replicas_num return self + def working_dir(self, working_dir): + self._update_container('workingDir', working_dir) + return self + def to_dict(self): return self._struct