Fix Formatting.

Author: suryans-commit

src/sagemaker/feature_store/feature_group.py

@@ -333,8 +333,7 @@ def _ingest_row(
                 (
                     FeatureValue(
                         feature_name=data_frame.columns[index - 1],
-                        value_as_string_list
-                        =IngestionManagerPandas._covert_feature_value_to_string_list(
+                        value_as_string_list=IngestionManagerPandas._covert_feature_to_string_list(
                             row[index]
                         ),
                     )
@@ -419,7 +418,7 @@ def _feature_value_is_not_none(
         return feature_value
 
     @staticmethod
-    def _covert_feature_value_to_string_list(feature_value: List[Any]):
+    def _covert_feature_to_string_list(feature_value: List[Any]):
         """Convert a list of feature values to a list of strings.
 
         Args:
@@ -1173,8 +1172,9 @@ def ingest(
             feature_group_name=self.name,
             feature_definitions=feature_definition_dict,
             sagemaker_session=self.sagemaker_session,
-            sagemaker_fs_runtime_client_config
-            =self.sagemaker_session.sagemaker_featurestore_runtime_client.meta.config,
+            sagemaker_fs_runtime_client_config=(
+                self.sagemaker_session.sagemaker_featurestore_runtime_client.meta.config
+            ),
             max_workers=max_workers,
             max_processes=max_processes,
             profile_name=profile_name,

tests/unit/test_session.py

@@ -5275,33 +5275,29 @@ def test_feature_group_put_record(sagemaker_session_with_fs_runtime_client):
         feature_group_name="MyFeatureGroup",
         record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
     )
-    assert (
-        sagemaker_session_with_fs_runtime_client.sagemaker_featurestore_runtime_client
-        .put_record.called_with(
-            FeatureGroupName="MyFeatureGroup",
-            record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
-        )
-    )
+    fs_client_mock = sagemaker_session_with_fs_runtime_client.sagemaker_featurestore_runtime_client
 
+    assert fs_client_mock.put_record.called_with(
+        FeatureGroupName="MyFeatureGroup",
+        record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
+    )
 
 
 def test_feature_group_put_record_with_ttl_and_target_stores(
-        sagemaker_session_with_fs_runtime_client,
+    sagemaker_session_with_fs_runtime_client,
 ):
     sagemaker_session_with_fs_runtime_client.put_record(
         feature_group_name="MyFeatureGroup",
         record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
         ttl_duration={"Unit": "Seconds", "Value": 123},
         target_stores=["OnlineStore", "OfflineStore"],
     )
-    assert (
-        sagemaker_session_with_fs_runtime_client.sagemaker_featurestore_runtime_client
-        .put_record.called_with(
-            FeatureGroupName="MyFeatureGroup",
-            record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
-            target_stores=["OnlineStore", "OfflineStore"],
-            ttl_duration={"Unit": "Seconds", "Value": 123},
-        )
+    fs_client_mock = sagemaker_session_with_fs_runtime_client.sagemaker_featurestore_runtime_client
+    assert fs_client_mock.put_record.called_with(
+        FeatureGroupName="MyFeatureGroup",
+        record=[{"FeatureName": "feature1", "ValueAsString": "value1"}],
+        target_stores=["OnlineStore", "OfflineStore"],
+        ttl_duration={"Unit": "Seconds", "Value": 123},
     )