def _get_auto_fill_queue_name()

in src/hyperpod_cli/commands/job.py [0:0]

• 8 lines of code
• 5 McCabe index (conditional complexity)

def _get_auto_fill_queue_name(namespace, scheduler_type):
    k8s_client = KubernetesClient()
    sm_managed_namespace = k8s_client.get_sagemaker_managed_namespace(namespace)
    queue_name = None
    # Provide queue name if not provided and scheduler type is SageMaker
    if sm_managed_namespace and scheduler_type == SchedulerType.SAGEMAKER.value:
        quota_allocation_id = sm_managed_namespace.metadata.labels[SAGEMAKER_QUOTA_ALLOCATION_LABEL]
        queue_name = HYPERPOD_NAMESPACE_PREFIX + quota_allocation_id + SAGEMAKER_MANAGED_LOCAL_QUEUE_SUFFIX
    return queue_name