Fixing enum accessor in UTs

PAIR-code · Sep 17, 2024 · daa6f69 · daa6f69
1 parent dc96cff
commit daa6f69
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lit_nlp/examples/gcp/model_server_test.py b/lit_nlp/examples/gcp/model_server_test.py
@@ -56,8 +56,8 @@ def setUpClass(cls):
         }],
     ),
     dict(
-        testcase_name=lit_gcp_constants.LlmHTTPEndpoints.TOKENIZE,
-        endpoint=f'/{lit_gcp_constants.LlmHTTPEndpoints.TOKENIZE}',
+        testcase_name=lit_gcp_constants.LlmHTTPEndpoints.TOKENIZE.value,
+        endpoint=f'/{lit_gcp_constants.LlmHTTPEndpoints.TOKENIZE.value}',
         expected=[{'tokens': ['test', 'output', 'text']}],
     ),
   )