Skip to content

Commit 4f37c21

Browse files
author
Nilesh
committed
chore: run black-format on feature store integ tests
1 parent 401fec5 commit 4f37c21

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

tests/integ/test_feature_store.py

+7-8
Original file line numberDiff line numberDiff line change
@@ -1074,21 +1074,20 @@ def cleanup_offline_store(feature_group: FeatureGroup, feature_store_session: Se
10741074
yield
10751075
finally:
10761076
feature_group_metadata = feature_group.describe()
1077-
feature_group_name = feature_group_metadata['FeatureGroupName']
1077+
feature_group_name = feature_group_metadata["FeatureGroupName"]
10781078
try:
1079-
s3_uri = feature_group_metadata['OfflineStoreConfig']['S3StorageConfig']['ResolvedOutputS3Uri']
1079+
s3_uri = feature_group_metadata["OfflineStoreConfig"]["S3StorageConfig"][
1080+
"ResolvedOutputS3Uri"
1081+
]
10801082
parsed_uri = urlparse(s3_uri)
10811083
bucket_name, prefix = parsed_uri.netloc, parsed_uri.path
1082-
prefix = prefix.strip('/')
1083-
prefix = prefix[:-5] if prefix.endswith('/data') else prefix
1084+
prefix = prefix.strip("/")
1085+
prefix = prefix[:-5] if prefix.endswith("/data") else prefix
10841086
region_name = feature_store_session.boto_session.region_name
10851087
s3_client = feature_store_session.boto_session.client(
10861088
service_name="s3", region_name=region_name
10871089
)
1088-
response = s3_client.list_objects_v2(
1089-
Bucket=bucket_name,
1090-
Prefix=prefix
1091-
)
1090+
response = s3_client.list_objects_v2(Bucket=bucket_name, Prefix=prefix)
10921091
files_in_folder = response["Contents"]
10931092
files_to_delete = []
10941093
for f in files_in_folder:

0 commit comments

Comments
 (0)