X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/0a2b2a825b451ea1a84ab3d58396c953981173cf..87e1614724027f6a9da3f3dfaa163bd8bea4ce62:/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 06b532ac56..02dd3dba06 100644
--- 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)