Make cmd nullable
@@ -7,8 +7,8 @@ from polyaxon_schemas.base import BaseConfig
class RunExecSchema(Schema):
- cmd = fields.Str()
image = fields.Str()
+ cmd = fields.Str(allow_none=True)
steps = fields.List(fields.Str(), allow_none=True)
env_vars = fields.List(fields.List(fields.Str(), validate=validate.Length(equal=2)),
allow_none=True)
@@ -31,7 +31,7 @@ class RunExecConfig(BaseConfig):
IDENTIFIER = 'run'
REDUCED_ATTRIBUTES = ['steps', 'env_vars', 'git']
- def __init__(self, cmd, image, steps=None, env_vars=None, git=None):
+ def __init__(self, image, cmd=None, steps=None, env_vars=None, git=None):
self.cmd = cmd
self.image = image
self.steps = steps
