ERROR:aws_xray_sdk.core.context:cannot find the current segment/subsegment, please make sure you have a segment open #452

import json
import os
import boto3
from aws_xray_sdk.core import xray_recorder
from aws_xray_sdk.core import patch_all
import logging
from botocore.exceptions import NoCredentialsError, PartialCredentialsError
from aws_xray_sdk import global_sdk_config
from boto3.s3.transfer import TransferConfig

patch_all()

logging.basicConfig(level='ERROR')
logging.getLogger('aws_xray_sdk.core').setLevel(logging.ERROR)

.
.
.
.
def generate_and_upload(init, count):
    session = boto3.Session(profile_name=aws_profile_name)
    s3_client = session.client("s3")

    # Create file path
    file_path = os.path.join("generated", "data.json")

    # Write data to the file
    with open(file_path, "w") as file:
        json.dump(
            generate_json_document(init, count),
            file,
            separators=(",", ":"),
            ensure_ascii=False,
        )
        
    xray_recorder.configure(service='data-generator')
    with xray_recorder.in_segment('generate_and_upload') as segment:
        trace_header = segment.trace_id
        print(f"Trace Header: {trace_header}")
        try:
            trnasver_config = TransferConfig(max_concurrency=1)
            s3_client.upload_file(
                file_path,
                f"{bucket}",
                f"{path}",
                ExtraArgs={"Metadata": {"X-Amzn-Trace-Id": trace_header}},
                Config=trnasver_config,
            )
            print(
                f"Uploaded"
            )
        finally:
            xray_recorder.end_segment()