Commit f7aef48c authored by Vít Starý Novotný's avatar Vít Starý Novotný
Browse files

Fix type check

parent 7770581d

ahisto_named_entity_search/recognition/model.py

+1 −0
Original line number Original line Diff line number Diff line
@@ -45,6 +45,7 @@ class NerModel: 
    @property

    @property

    def model(self) -> AutoModelForTokenClassification:

    def model(self) -> AutoModelForTokenClassification:

        model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_basename)

        model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_basename)

        return model





    def __str__(self) -> str:

    def __str__(self) -> str:

        return self.model_name_or_basename

        return self.model_name_or_basename