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" )
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")
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
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"