IG-13196: Set working directory to /User (#8)

* Set working directory to /User * bump version to 0.0.8 * fix yams * add "working_dir" function
v3io · Sep 5, 2019 · 13ac31a · 13ac31a
1 parent c58fdc9
commit 13ac31a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git 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
@@ -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