Fix unit tests

icywang86rui · icywang86rui · commit baa3b988c00c · 2018-08-15T13:46:48.000-07:00
diff --git a/tests/unit/test_chainer.py b/tests/unit/test_chainer.py
@@ -122,7 +122,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
@@ -113,7 +113,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
`@@ -122,7 +122,7 @@ def _create_train_job(version):`
`122`	`122`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`123`	`123`	`},`
`124`	`124`	`'tags': None,`
`125`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`125`	`+ 'vpc_config': None`
`126`	`126`	`}`
`127`	`127`
`128`	`128`
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
`@@ -113,7 +113,7 @@ def _create_train_job(version):`
`113`	`113`	`'MaxRuntimeInSeconds': 24 * 60 * 60`
`114`	`114`	`},`
`115`	`115`	`'tags': None,`
`116`		`- 'vpc_config': {'SecurityGroupIds': None, 'Subnets': None}`
	`116`	`+ 'vpc_config': None`
`117`	`117`	`}`
`118`	`118`
`119`	`119`
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`