fix more flake8

Author: mollyheamazon

src/sagemaker/feature_store/dataset_builder.py

@@ -929,7 +929,7 @@ def _construct_query_string(self, base: FeatureGroupToBeMerged) -> str:
                 selected_features += ", "
                 selected_features += ", ".join(
                     [
-                        f'fg_{i}."{feature_name}" as "{feature_name}.{(i+1)}"'
+                        f'fg_{i}."{feature_name}" as "{feature_name}.{(i + 1)}"'
                         for feature_name in feature_group.projected_feature_names
                     ]
                 )

tests/unit/test_predictor_async.py

@@ -233,7 +233,7 @@ def test_async_predict_call_verify_exceptions():
     with pytest.raises(
         PollingTimeoutError,
         match=f"No result at {ASYNC_OUTPUT_LOCATION} after polling for "
-        f"{DEFAULT_WAITER_CONFIG.delay*DEFAULT_WAITER_CONFIG.max_attempts}"
+        f"{DEFAULT_WAITER_CONFIG.delay * DEFAULT_WAITER_CONFIG.max_attempts}"
         f" seconds. Inference could still be running",
     ):
         predictor_async.predict(input_path=input_location, waiter_config=DEFAULT_WAITER_CONFIG)
@@ -253,7 +253,7 @@ def test_async_predict_call_verify_exceptions_with_null_failure_path():
     with pytest.raises(
         PollingTimeoutError,
         match=f"No result at {ASYNC_OUTPUT_LOCATION} after polling for "
-        f"{DEFAULT_WAITER_CONFIG.delay*DEFAULT_WAITER_CONFIG.max_attempts}"
+        f"{DEFAULT_WAITER_CONFIG.delay * DEFAULT_WAITER_CONFIG.max_attempts}"
         f" seconds. Inference could still be running",
     ):
         predictor_async.predict(input_path=input_location, waiter_config=DEFAULT_WAITER_CONFIG)