code: def create_security_group(security_group_name: str, vpc_id: str):
      client = boto3.client("ec2")
      security_group = client.create_security_group(
          GroupName=security_group_name,
          Description="Created for DMS system test",
          VpcId=vpc_id,
      )
      client.get_waiter("security_group_exists").wait(
          GroupIds=[security_group["GroupId"]],
      )
      client.authorize_security_group_ingress(
          GroupId=security_group["GroupId"],
          IpPermissions=[SG_IP_PERMISSION],
      )
  
      return security_group["GroupId"]
  
imports: from __future__ import annotations
  import json
  from datetime import datetime
  from typing import cast
  import boto3
  from sqlalchemy import Column, MetaData, String, Table, create_engine
  from airflow.decorators import task
  from airflow.models.baseoperator import chain
  from airflow.models.dag import DAG
  from airflow.providers.amazon.aws.operators.dms import DmsCreateTaskOperator, DmsDeleteTaskOperator, DmsDescribeTasksOperator, DmsStartTaskOperator, DmsStopTaskOperator
  from airflow.providers.amazon.aws.operators.rds import RdsCreateDbInstanceOperator, RdsDeleteDbInstanceOperator
  from airflow.providers.amazon.aws.operators.s3 import S3CreateBucketOperator, S3DeleteBucketOperator
  from airflow.providers.amazon.aws.sensors.dms import DmsTaskBaseSensor, DmsTaskCompletedSensor
  from airflow.utils.trigger_rule import TriggerRule
  from tests.system.providers.amazon.aws.utils import ENV_ID_KEY, SystemTestContextBuilder
  from tests.system.providers.amazon.aws.utils.ec2 import get_default_vpc_id
  from tests.system.utils import get_test_run
label_desc: Use image node_id 2 for calling a global function with example usage: create_security_group(security_group_name, vpc_id) and returns: security_group