code: def cleanup_offline_store(
      feature_group: FeatureGroup, feature_store_session: Session
  ):
      try:
          yield
      finally:
          feature_group_metadata = feature_group.describe()
          feature_group_name = feature_group_metadata[
              "FeatureGroupName"
          ]
          try:
              s3_uri = feature_group_metadata["OfflineStoreConfig"][
                  "S3StorageConfig"
              ]["ResolvedOutputS3Uri"]
              parsed_uri = urlparse(s3_uri)
              bucket_name, prefix = parsed_uri.netloc, parsed_uri.path
              prefix = prefix.strip("/")
              prefix = (
                  prefix[:-5] if prefix.endswith("/data") else prefix
              )
              region_name = (
                  feature_store_session.boto_session.region_name
              )
              s3_client = feature_store_session.boto_session.client(
                  service_name="s3", region_name=region_name
              )
              response = s3_client.list_objects_v2(
                  Bucket=bucket_name, Prefix=prefix
              )
              files_in_folder = response["Contents"]
              files_to_delete = []
              for f in files_in_folder:
                  files_to_delete.append({"Key": f["Key"]})
              s3_client.delete_objects(
                  Bucket=bucket_name,
                  Delete={"Objects": files_to_delete},
              )
          except Exception as e:
              raise RuntimeError(
                  f"Failed to delete data for feature_group {feature_group_name}",
                  e,
              )
  
imports: from __future__ import absolute_import
  import datetime
  import json
  import time
  import dateutil.parser
  from contextlib import contextmanager
  import boto3
  import numpy
  import pandas
  import pytest
  from pandas import DataFrame
  from sagemaker.feature_store.feature_utils import get_feature_group_as_dataframe
  from sagemaker.feature_store.feature_definition import FractionalFeatureDefinition, StringFeatureDefinition, ListCollectionType
  from sagemaker.feature_store.feature_group import FeatureGroup
  from sagemaker.feature_store.feature_store import FeatureStore
  from sagemaker.feature_store.inputs import FeatureValue, FeatureParameter, TableFormatEnum, Filter, ResourceEnum, Identifier, DeletionModeEnum, TtlDuration, OnlineStoreConfigUpdate, OnlineStoreStorageTypeEnum
  from sagemaker.feature_store.dataset_builder import JoinTypeEnum
  from sagemaker.session import get_execution_role, Session
  from tests.integ.timeout import timeout
  from urllib.parse import urlparse
label_desc: Use image node_id 42 for calling a global function with example usage: cleanup_offline_store(feature_group, feature_store_session) without return types