def main()

in src/python-sdk-v2/create_batch_deployment.py [0:0]


def main():
    args = parse_args()
    print(args)
    
    credential = DefaultAzureCredential()
    try:
        ml_client = MLClient.from_config(credential, path='config.json')

    except Exception as ex:
        print("HERE IN THE EXCEPTION BLOCK")
        print(ex)

    batch_deployment = BatchDeployment(
        name=args.deployment_name,
        description=args.description,
        endpoint_name=args.endpoint_name,
        model=args.model_path,
        compute=args.compute,
        instance_count=args.instance_count,
        max_concurrency_per_instance=args.max_concurrency_per_instance,
        mini_batch_size=args.mini_batch_size,
        output_action=BatchDeploymentOutputAction.APPEND_ROW,
        output_file_name=args.output_file_name
    )

    deployment_job = ml_client.batch_deployments.begin_create_or_update(
        deployment=batch_deployment
    )
    deployment_job.wait()

    batch_endpoint = ml_client.batch_endpoints.get(args.endpoint_name)
    batch_endpoint.defaults.deployment_name = batch_deployment.name

    endpoint_update_job = ml_client.batch_endpoints.begin_create_or_update(
        batch_endpoint
    )
    endpoint_update_job.wait()