Fix seq class initialization

Signed-off-by: Alex-Brooks <[email protected]>
opendatahub-io · Sep 12, 2023 · fca1afa · fca1afa
1 parent 0161d72
commit fca1afa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/caikit_nlp/resources/pretrained_model/hf_auto_seq_classifier.py b/caikit_nlp/resources/pretrained_model/hf_auto_seq_classifier.py
@@ -55,7 +55,7 @@ def bootstrap(cls, *args, **kwargs) -> "HFAutoSequenceClassifier":
  return super().bootstrap(*args, return_dict=True, **kwargs)
 
  @staticmethod
- def build_task_tokenize_closure(*args, **kwargs) -> Tuple[Callable, bool]:
+ def tokenize_function(*args, **kwargs) -> Tuple[Callable, bool]:
  raise NotImplementedError(
- "Tokenize func builder not implemented for sequence classifier"
+ "Tokenize func not implemented for sequence classifier"
  )