Fix unit tests

icywang86rui · icywang86rui · commit 149ac7a03e2e · 2018-08-21T13:09:59.000-07:00
diff --git a/tests/unit/test_chainer.py b/tests/unit/test_chainer.py
@@ -121,7 +121,7 @@ def _create_train_job(version):
             'MaxRuntimeInSeconds': 24 * 60 * 60
         },
         'tags': None,
-        'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}
+        'vpc_config': None
     }
 
 
diff --git a/tests/unit/test_estimator.py b/tests/unit/test_estimator.py
@@ -658,7 +658,7 @@ def test_unsupported_type_in_dict():
     },
     'stop_condition': {'MaxRuntimeInSeconds': 86400},
     'tags': None,
-    'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}
+    'vpc_config': None
 }
 
 HYPERPARAMS = {'x': 1, 'y': 'hello'}
diff --git a/tests/unit/test_mxnet.py b/tests/unit/test_mxnet.py
@@ -95,7 +95,7 @@ def _create_train_job(version):
             'MaxRuntimeInSeconds': 24 * 60 * 60
         },
         'tags': None,
-        'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}
+        'vpc_config': None
     }
 
 
diff --git a/tests/unit/test_pytorch.py b/tests/unit/test_pytorch.py
@@ -112,7 +112,7 @@ def _create_train_job(version):
             'MaxRuntimeInSeconds': 24 * 60 * 60
         },
         'tags': None,
-        'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}
+        'vpc_config': None
     }
 
 
diff --git a/tests/unit/test_tf_estimator.py b/tests/unit/test_tf_estimator.py
@@ -103,7 +103,7 @@ def _create_train_job(tf_version):
             'MaxRuntimeInSeconds': 24 * 60 * 60
         },
         'tags': None,
-        'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}
+        'vpc_config': None
     }
 
 

Original file line number	Diff line number	Diff line change
`@@ -121,7 +121,7 @@ def _create_train_job(version):`
`121`	`121`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`122`	`122`	`},`
`123`	`123`	`'tags': None,`
`124`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`124`	`+ 'vpc_config': None`
`125`	`125`	`}`
`126`	`126`
`127`	`127`
Original file line number	Diff line number	Diff line change
`@@ -658,7 +658,7 @@ def test_unsupported_type_in_dict():`
`658`	`658`	`},`
`659`	`659`	`'stop_condition': {'MaxRuntimeInSeconds': 86400},`
`660`	`660`	`'tags': None,`
`661`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`661`	`+ 'vpc_config': None`
`662`	`662`	`}`
`663`	`663`
`664`	`664`	`HYPERPARAMS = {'x': 1, 'y': 'hello'}`
Original file line number	Diff line number	Diff line change
`@@ -95,7 +95,7 @@ def _create_train_job(version):`
`95`	`95`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`96`	`96`	`},`
`97`	`97`	`'tags': None,`
`98`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`98`	`+ 'vpc_config': None`
`99`	`99`	`}`
`100`	`100`
`101`	`101`
Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def _create_train_job(version):`
`112`	`112`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`113`	`113`	`},`
`114`	`114`	`'tags': None,`
`115`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`115`	`+ 'vpc_config': None`
`116`	`116`	`}`
`117`	`117`
`118`	`118`
Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def _create_train_job(tf_version):`
`103`	`103`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`104`	`104`	`},`
`105`	`105`	`'tags': None,`
`106`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`106`	`+ 'vpc_config': None`
`107`	`107`	`}`
`108`	`108`
`109`	`109`