/ src / evidently / legacy / descriptors / trigger_words_presence_descriptor.py
trigger_words_presence_descriptor.py
 1  from typing import List
 2  
 3  from evidently.legacy.features import trigger_words_presence_feature
 4  from evidently.legacy.features.generated_features import FeatureDescriptor
 5  from evidently.legacy.features.generated_features import GeneratedFeature
 6  
 7  
 8  class TriggerWordsPresence(FeatureDescriptor):
 9      class Config:
10          type_alias = "evidently:descriptor:TriggerWordsPresence"
11  
12      words_list: List[str]
13      lemmatize: bool = True
14  
15      def feature(self, column_name: str) -> GeneratedFeature:
16          return trigger_words_presence_feature.TriggerWordsPresent(
17              column_name,
18              self.words_list,
19              self.lemmatize,
20              self.display_name,
21          )