Python DefaultV1Recipe._from_registry示例

示例#1

def test_retrieve_not_registered_class():
    class NotRegisteredClass:
        pass

    with pytest.raises(InvalidConfigException):
        # noinspection PyTypeChecker
        DefaultV1Recipe._from_registry(NotRegisteredClass.__name__)

示例#2

def test_register_component_with_multiple_types():
    @DefaultV1Recipe.register(
        [
            DefaultV1Recipe.ComponentType.MESSAGE_TOKENIZER,
            DefaultV1Recipe.ComponentType.MODEL_LOADER,
        ],
        is_trainable=True,
        model_from="Herman",
    )
    class MyClassGraphComponent(GraphComponent):
        @classmethod
        def create(
            cls,
            config: Dict[Text, Any],
            model_storage: ModelStorage,
            resource: Resource,
            execution_context: ExecutionContext,
        ) -> GraphComponent:
            return cls()

    assert DefaultV1Recipe._from_registry(
        MyClassGraphComponent.__name__) == DefaultV1Recipe.RegisteredComponent(
            MyClassGraphComponent,
            {
                DefaultV1Recipe.ComponentType.MESSAGE_TOKENIZER,
                DefaultV1Recipe.ComponentType.MODEL_LOADER,
            },
            True,
            "Herman",
        )
    assert MyClassGraphComponent()