Update test_feature_store.py

Author: can-sun

tests/integ/test_feature_store.py

@@ -142,8 +142,7 @@ def create_table_ddl():
         "  STORED AS\n"
         "  INPUTFORMAT 'parquet.hive.DeprecatedParquetInputFormat'\n"
         "  OUTPUTFORMAT 'parquet.hive.DeprecatedParquetOutputFormat'\n"
-        "LOCATION 's3://sagemaker-test-featurestore-{region}-{account}"
-        "/{account}/sagemaker/us-east-2/offline-store/{feature_group_name}'"
+        "LOCATION '{resolved_output_s3_uri}'"
     )
 
 
@@ -180,7 +179,8 @@ def test_create_feature_store(
 ):
     feature_group = FeatureGroup(name=feature_group_name, sagemaker_session=feature_store_session)
     feature_group.load_feature_definitions(data_frame=pandas_data_frame)
-
+    resolved_output_s3_uri = feature_group.describe().get("OfflineStoreConfig").get("S3StorageConfig").get("ResolvedOutputS3Uri")
+    
     with cleanup_feature_group(feature_group):
         output = feature_group.create(
             s3_uri=offline_store_s3_uri,
@@ -225,6 +225,7 @@ def test_create_feature_store(
                 feature_group_name=feature_group_name,
                 region=feature_store_session.boto_session.region_name,
                 account=feature_store_session.account_id(),
+                resolved_output_s3_uri=resolved_output_s3_uri
             )
             == feature_group.as_hive_ddl()
         )