From d2f738793c8ec5ad9b66aeec78fac74a76b37100 Mon Sep 17 00:00:00 2001
From: Sebastian Schmidt <mrschmidt@google.com>
Date: Tue, 20 Dec 2022 15:15:24 -0800
Subject: [PATCH] Use uppercase options name for "delegate"

PiperOrigin-RevId: 496764089
---
 .../tasks/web/components/processors/base_options.test.ts    | 6 +++---
 mediapipe/tasks/web/components/processors/base_options.ts   | 2 +-
 mediapipe/tasks/web/core/task_runner_options.d.ts           | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/mediapipe/tasks/web/components/processors/base_options.test.ts b/mediapipe/tasks/web/components/processors/base_options.test.ts
index 46c2277e..6d58be68 100644
--- a/mediapipe/tasks/web/components/processors/base_options.test.ts
+++ b/mediapipe/tasks/web/components/processors/base_options.test.ts
@@ -86,7 +86,7 @@ describe('convertBaseOptionsToProto()', () => {
   it('can enable CPU delegate', async () => {
     const baseOptionsProto = await convertBaseOptionsToProto({
       modelAssetBuffer: new Uint8Array(mockBytes),
-      delegate: 'cpu',
+      delegate: 'CPU',
     });
     expect(baseOptionsProto.toObject()).toEqual(mockBytesResult);
   });
@@ -94,7 +94,7 @@ describe('convertBaseOptionsToProto()', () => {
   it('can enable GPU delegate', async () => {
     const baseOptionsProto = await convertBaseOptionsToProto({
       modelAssetBuffer: new Uint8Array(mockBytes),
-      delegate: 'gpu',
+      delegate: 'GPU',
     });
     expect(baseOptionsProto.toObject()).toEqual({
       ...mockBytesResult,
@@ -117,7 +117,7 @@ describe('convertBaseOptionsToProto()', () => {
   it('can reset delegate', async () => {
     let baseOptionsProto = await convertBaseOptionsToProto({
       modelAssetBuffer: new Uint8Array(mockBytes),
-      delegate: 'gpu',
+      delegate: 'GPU',
     });
     // Clear backend
     baseOptionsProto =
diff --git a/mediapipe/tasks/web/components/processors/base_options.ts b/mediapipe/tasks/web/components/processors/base_options.ts
index 16d56226..97b62b78 100644
--- a/mediapipe/tasks/web/components/processors/base_options.ts
+++ b/mediapipe/tasks/web/components/processors/base_options.ts
@@ -71,7 +71,7 @@ async function configureExternalFile(
 /** Configues the `acceleration` option. */
 function configureAcceleration(options: BaseOptions, proto: BaseOptionsProto) {
   const acceleration = proto.getAcceleration() ?? new Acceleration();
-  if (options.delegate === 'gpu') {
+  if (options.delegate === 'GPU') {
     acceleration.setGpu(new InferenceCalculatorOptions.Delegate.Gpu());
   } else {
     acceleration.setTflite(new InferenceCalculatorOptions.Delegate.TfLite());
diff --git a/mediapipe/tasks/web/core/task_runner_options.d.ts b/mediapipe/tasks/web/core/task_runner_options.d.ts
index aa0b4a02..5f23cd4b 100644
--- a/mediapipe/tasks/web/core/task_runner_options.d.ts
+++ b/mediapipe/tasks/web/core/task_runner_options.d.ts
@@ -31,7 +31,7 @@ export declare interface BaseOptions {
   modelAssetBuffer?: Uint8Array|undefined;
 
   /** Overrides the default backend to use for the provided model. */
-  delegate?: 'cpu'|'gpu'|undefined;
+  delegate?: 'CPU'|'GPU'|undefined;
 }
 
 /** Options to configure MediaPipe Tasks in general. */