Skip to content

Commit 81c1dbf

Browse files
fix: DEV-1959: Fix ner example 'TransformersBasedTagger' object has no attribute '_tokenizer' (#112)
* fix: DEV-1959: Fix ner example 'TransformersBasedTagger' object has no attribute '_tokenizer' * Update ner.py Co-authored-by: Max Tkachenko <[email protected]>
1 parent 8d8198b commit 81c1dbf

File tree

1 file changed

+3
-0
lines changed
  • label_studio_ml/examples/ner

1 file changed

+3
-0
lines changed

label_studio_ml/examples/ner/ner.py

+3
Original file line numberDiff line numberDiff line change
@@ -365,6 +365,9 @@ def load(self, train_output):
365365
self._model_type, self._tokenizer, self._pad_token_label_id)
366366

367367
def predict(self, tasks, **kwargs):
368+
if not hasattr(self, "_tokenizer"):
369+
logger.error("No model loaded! Please train model before getting predictions!")
370+
return {}
368371
texts = [task['data'][self.value] for task in tasks]
369372
predict_set = SpanLabeledTextDataset(texts, tokenizer=self._tokenizer, **self._dataset_params_dict)
370373
from_name = self.from_name

0 commit comments

Comments
 (0)