Ejemplos de EntityExtractor.clean_up_entities en Python

Ejemplo n.º 1

Archivo: test_extractor.py Proyecto: varunkothamachu/rasa

def test_convert_tags_to_entities(
    entities: List[Dict[Text, Any]],
    keep: bool,
    expected_entities: List[Dict[Text, Any]],
):
    extractor = EntityExtractor()

    updated_entities = extractor.clean_up_entities(entities, keep)

    assert updated_entities == expected_entities

Ejemplo n.º 2

Archivo: test_extractor.py Proyecto: mufasil/rasa-for-botfront

def test_clean_up_entities(
    text: Text,
    tokens: List[Token],
    entities: List[Dict[Text, Any]],
    keep: bool,
    expected_entities: List[Dict[Text, Any]],
):
    extractor = EntityExtractor()

    message = Message(text)
    message.set("tokens", tokens)

    updated_entities = extractor.clean_up_entities(message, entities, keep)

    assert updated_entities == expected_entities