fix: use SecurityConfig dict key (#2064)

ajaykarpur · ahsan-z-khan · web-flow · commit d2c317dc167f · 2021-01-05T11:59:45.000-08:00
* fix: use SecurityConfig dict key

* update unit test

Co-authored-by: Ahsan Khan &lt;ahsan.al.zaki@gmail.com&gt;
diff --git a/src/sagemaker/feature_store/inputs.py b/src/sagemaker/feature_store/inputs.py
@@ -103,7 +103,7 @@ def to_dict(self) -> Dict[str, Any]:
         """
         return Config.construct_dict(
             EnableOnlineStore=self.enable_online_store,
-            OnlineStoreSecurityConfig=self.online_store_security_config,
+            SecurityConfig=self.online_store_security_config,
         )
 
 
diff --git a/tests/unit/sagemaker/feature_store/test_inputs.py b/tests/unit/sagemaker/feature_store/test_inputs.py
@@ -48,7 +48,7 @@ def test_online_store_config():
     assert ordered(config_with_kms.to_dict()) == ordered(
         {
             "EnableOnlineStore": True,
-            "OnlineStoreSecurityConfig": {
+            "SecurityConfig": {
                 "KmsKeyId": "kms",
             },
         }

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def to_dict(self) -> Dict[str, Any]:`
`103`	`103`	`"""`
`104`	`104`	`return Config.construct_dict(`
`105`	`105`	`EnableOnlineStore=self.enable_online_store,`
`106`		`- OnlineStoreSecurityConfig=self.online_store_security_config,`
	`106`	`+ SecurityConfig=self.online_store_security_config,`
`107`	`107`	`)`
`108`	`108`
`109`	`109`
Original file line number	Diff line number	Diff line change
`@@ -48,7 +48,7 @@ def test_online_store_config():`
`48`	`48`	`assert ordered(config_with_kms.to_dict()) == ordered(`
`49`	`49`	`{`
`50`	`50`	`"EnableOnlineStore": True,`
`51`		`- "OnlineStoreSecurityConfig": {`
	`51`	`+ "SecurityConfig": {`
`52`	`52`	`"KmsKeyId": "kms",`
`53`	`53`	`},`
`54`	`54`	`}`