@@ -28,6 +28,7 @@ class TestExperimentsCreateSingleNode(object):
2828 "--container" , "testContainer" ,
2929 "--machineType" , "testType" ,
3030 "--command" , "testCommand" ,
31+ "--workspaceUrl" , "some-workspace" ,
3132 ]
3233 FULL_OPTIONS_COMMAND = [
3334 "experiments" , "create" , "singlenode" ,
@@ -56,6 +57,7 @@ class TestExperimentsCreateSingleNode(object):
5657 "machineType" : u"testType" ,
5758 "command" : u"testCommand" ,
5859 "experimentTypeId" : constants .ExperimentType .SINGLE_NODE ,
60+ "workspaceUrl" : u"some-workspace" ,
5961 }
6062 FULL_OPTIONS_REQUEST = {
6163 "name" : u"exp1" ,
@@ -148,6 +150,7 @@ class TestExperimentsCreateMultiNode(object):
148150 "--parameterServerCommand" , "ls" ,
149151 "--parameterServerCount" , 2 ,
150152 "--workerContainerUser" , "usr" ,
153+ "--workspaceUrl" , "some-workspace" ,
151154 ]
152155 FULL_OPTIONS_COMMAND = [
153156 "experiments" , "create" , "multinode" ,
@@ -190,6 +193,7 @@ class TestExperimentsCreateMultiNode(object):
190193 "parameterServerCommand" : u"ls" ,
191194 "parameterServerCount" : 2 ,
192195 "workerContainerUser" : u"usr" ,
196+ "workspaceUrl" : "some-workspace" ,
193197 }
194198 FULL_OPTIONS_REQUEST = {
195199 "name" : u"multinode_mpi" ,
@@ -262,6 +266,7 @@ class TestExperimentsCreateAndStartSingleNode(TestExperimentsCreateSingleNode):
262266 "--container" , "testContainer" ,
263267 "--machineType" , "testType" ,
264268 "--command" , "testCommand" ,
269+ "--workspaceUrl" , "some-workspace" ,
265270 ]
266271 FULL_OPTIONS_COMMAND = [
267272 "experiments" , "createAndStart" , "singlenode" ,
@@ -302,6 +307,7 @@ class TestExperimentsCreateAndStartMultiNode(TestExperimentsCreateMultiNode):
302307 "--parameterServerCommand" , "ls" ,
303308 "--parameterServerCount" , 2 ,
304309 "--workerContainerUser" , "usr" ,
310+ "--workspaceUrl" , "some-workspace" ,
305311 ]
306312 FULL_OPTIONS_COMMAND = [
307313 "experiments" , "createAndStart" , "multinode" ,
0 commit comments