Ejemplo n.º 1
0
def execute(args: argparse.Namespace):
    config_file = os.path.join(args.dir, args.c)
    config = get_config(config_file)

    with tempfile.TemporaryDirectory(dir=os.path.abspath(args.dir)) as td:
        if config.deployment.mode == "batch":
            builder: ImageBuilder = BatchImageBuilder(source_dir=args.dir,
                                                      target_dir=td)
        elif config.deployment.mode == "http":
            builder = FlaskImageBuilder(source_dir=args.dir, target_dir=td)
        else:
            raise NotImplementedError(
                f"Currently no support for mode {config.deployment.mode}")

        builder.generate_wrapper_files(config)
        builder.copy_client_files_to_build_dir()
        builder.build_image(config)
Ejemplo n.º 2
0
def test_get_config_default(tmp_path, generate_config) -> None:

    file = tmp_path / "file"
    file.write_text(generate_config())
    config = get_config(file)
    assert isinstance(config, NervosumConfig)
    assert config.name == "a_name"
    assert config.deployment.mode == "batch"
    assert config.src == "a_src"
    assert config.tag == "a_tag"
    assert config.deployment.platform_tag is None
    assert isinstance(config.interface, Interface)
    assert config.interface.model_module == "a_module"
    assert config.interface.model_class == "a_class"
    assert config.requirements == "a_requirements_file"
    assert isinstance(config.input_schema, List)
    assert config.input_schema[0].name == "field"
    assert config.input_schema[0].type == "a_type"
    assert config.output_schema[0].name == "prediction"
    assert config.output_schema[0].type == "a_type"
Ejemplo n.º 3
0
def test_get_config_mode_unsupported(tmp_path, generate_config) -> None:
    file = tmp_path / "file"
    file.write_text(generate_config(mode="unsupported"))
    with pytest.raises(ValidationError):
        get_config(file)
Ejemplo n.º 4
0
def test_get_config_mode_http(tmp_path, generate_config) -> None:
    file = tmp_path / "file"
    file.write_text(generate_config(mode="http"))
    config = get_config(file)
    assert config.deployment.mode == "http"