7454: Add 'size' to arvados_create_kwargs so that it can be included in CustomData.

[arvados.git] / services / nodemanager / arvnodeman / computenode / driver / __init__.py

diff --git a/services/nodemanager/arvnodeman/computenode/driver/__init__.py b/services/nodemanager/arvnodeman/computenode/driver/__init__.py
index 06b532ac5618d79915478930d05072278ad3753f..02dd3dba06903286fbb41efcb864e81741aca84e 100644 (file)
--- a/services/nodemanager/arvnodeman/computenode/driver/__init__.py
+++ b/services/nodemanager/arvnodeman/computenode/driver/__init__.py
@@ -91,7 +91,7 @@ class BaseComputeNodeDriver(object):
     def list_nodes(self):
         return self.real.list_nodes(**self.list_kwargs)
 
-    def arvados_create_kwargs(self, arvados_node):
+    def arvados_create_kwargs(self, arvados_node, size):
         """Return dynamic keyword arguments for create_node.
 
         Subclasses must override this method.  It should return a dictionary
@@ -116,7 +116,7 @@ class BaseComputeNodeDriver(object):
 
     def create_node(self, size, arvados_node):
         kwargs = self.create_kwargs.copy()
-        kwargs.update(self.arvados_create_kwargs(arvados_node))
+        kwargs.update(self.arvados_create_kwargs(arvados_node, size))
         kwargs['size'] = size
         return self.real.create_node(**kwargs)