fix: black checks

Author: JoseJuan98

src/sagemaker/feature_store/feature_utils.py

@@ -46,9 +46,7 @@ def get_session_from_role(region: str, assume_role: str = None) -> Session:
 
     # It will try to assume the role specified
     if assume_role:
-        sts = boto_session.client(
-            "sts", region_name=region
-        )
+        sts = boto_session.client("sts", region_name=region)
 
         credentials = sts.assume_role(
             RoleArn=assume_role, RoleSessionName="SagemakerExecution"

tests/integ/test_feature_store.py

@@ -1087,6 +1087,7 @@ def test_get_feature_group_with_role_region(
             event_time_feature_name="feature3",
             latest_ingestion=True,
             athena_bucket=f"{offline_store_s3_uri}/query",
+            verbose=False,
         )
 
         assert not dataset.empty
@@ -1124,6 +1125,7 @@ def test_get_feature_group_with_session(
             event_time_feature_name="feature3",
             latest_ingestion=True,
             athena_bucket=f"{offline_store_s3_uri}/query",
+            verbose=False,
             low_memory=False,
         )  # Using kwargs to pass a parameter to pandas.read_csv