Esempio n. 1
0
def test_overwrite_model_server_url_with_no_model_endpoint():
    endpoints = AvailableEndpoints()
    x._overwrite_endpoints_for_local_x(endpoints, "test", "http://localhost")
    assert (
        endpoints.model.url
        == "http://localhost/projects/default/models/tags/production"
    )
Esempio n. 2
0
def test_overwrite_model_server_url():
    endpoint_config = EndpointConfig(
        url="http://*****:*****@latest")
    endpoints = AvailableEndpoints(model=endpoint_config)
    x._overwrite_endpoints_for_local_x(endpoints, "test", "http://localhost")
    assert (endpoints.model.url ==
            "http://localhost/projects/default/models/tag/production")
Esempio n. 3
0
def test_default_wait_time_between_pulls():
    endpoint_config = EndpointConfig(url="http://*****:*****@latest")
    endpoints = AvailableEndpoints(model=endpoint_config)
    x._overwrite_endpoints_for_local_x(endpoints, "test", "http://localhost")
    assert endpoints.model.kwargs["wait_time_between_pulls"] == 2
Esempio n. 4
0
def test_default_model_server_url():
    endpoint_config = EndpointConfig()
    endpoints = AvailableEndpoints(model=endpoint_config)
    x._overwrite_endpoints_for_local_x(endpoints, "test", "http://localhost")
    assert endpoints.model.url == "http://localhost/models/tags/production"