Python new_model_serverの例

コード例 #1

ファイル: test_code_to_func.py プロジェクト: zhuohuwu0603/mlrun

def test_nuclio_nb():
    fn = new_model_server('iris-srv',
                          filename='{}/xgb_serving.ipynb'.format(examples_path),
                          models={'iris_v1': 'xyz'},
                          model_class='XGBoostModel')
    assert fn.kind == 'remote', 'kind not set, test failed'
    assert fn.spec.base_spec, 'base_spec not found'

コード例 #2

def test_nuclio_nb():
    fn = new_model_server(
        "iris-srv",
        filename=f"{examples_path}/xgb_serving.ipynb",
        models={"iris_v1": "xyz"},
        model_class="XGBoostModel",
    )
    assert fn.kind == "remote", "kind not set, test failed"
    assert fn.spec.base_spec, "base_spec not found"