From cb42f148d9d9b27af8ee08e3a154a6029087a503 Mon Sep 17 00:00:00 2001 From: Konstantin Korotaev <42615530+KonstantinKorotaev@users.noreply.github.com> Date: Wed, 1 Jun 2022 14:21:03 +0300 Subject: [PATCH] fix: DEV-2502: Add stub for fit method in ner example --- label_studio_ml/examples/ner/ner.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/label_studio_ml/examples/ner/ner.py b/label_studio_ml/examples/ner/ner.py index ea2ac6bac..a652989cf 100644 --- a/label_studio_ml/examples/ner/ner.py +++ b/label_studio_ml/examples/ner/ner.py @@ -4,6 +4,7 @@ import os import io import logging +import random from functools import partial from itertools import groupby @@ -27,7 +28,6 @@ from label_studio_ml.model import LabelStudioMLBase from utils import calc_slope - logger = logging.getLogger(__name__) @@ -325,6 +325,9 @@ def get_padding_function(cls, model_type, tokenizer, pad_token_label_id): pad_token_label_id=pad_token_label_id ) + def fit(self, completions, workdir=None, **kwargs): + return {'random': random.randint(1, 10)} + class TransformersBasedTagger(LabelStudioMLBase):