pod_template = PodTemplate(
    primary_container_name="primary",
    annotations={
            "<http://scheduling.volcano.sh/queue-name|scheduling.volcano.sh/queue-name>": "training"
        },
    pod_spec=V1PodSpec(
        scheduler_name="volcano",  # Specify Volcano scheduler
        containers=[
            V1Container(
                name="primary",
                resources=V1ResourceRequirements(
                    requests={
                        "cpu": "1",
                        "memory": "1Gi"
                    },
                    limits={
                        "cpu": "2",
                        "memory": "2Gi"
                    }
                )
            )
        ],
        # Specify the Volcano queue name using the correct annotation
    )
)