def test_fail_upload_model(): headers = {'Authorization': get_user_token()['id_token']} params = { 'model_name': MODEL_NAME, 'model_version': 1, 'file_path': os.path.abspath('non-existing-model.pb'), } url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}" with pytest.raises(Exception): upload_model(url, params, headers, UPLOAD_PART_SIZE)
def test_upload_model(): try: response = requests.get('https://storage.googleapis.com/inference-eu/models_zoo/' 'resnet_V1_50/saved_model/saved_model.pb') with open('saved_model.pb', 'wb') as f: f.write(response.content) headers = {'Authorization': get_user_token()['id_token']} params = { 'model_name': MODEL_NAME, 'model_version': 1, 'file_path': os.path.abspath('saved_model.pb'), } url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}" upload_model(url, params, headers, 30) os.remove('saved_model.pb') except Exception as e: pytest.fail(f"Unexpected error during upload test {e}")
def test_upload_models(): try: headers = {'Authorization': get_user_token()['id_token']} params = { 'model_name': MODEL_NAME, 'model_version': 1, 'file_path': os.path.abspath('saved_model.pb'), } url = f"{MANAGEMENT_API_URL}/tenants/{TENANT_NAME}" # resnet_v1_50 upload download_saved_model_from_path( 'https://storage.googleapis.com/inference-eu/models_zoo/' 'resnet_V1_50/saved_model/saved_model.pb') upload_model(url, params, headers, UPLOAD_PART_SIZE) os.remove('saved_model.pb') # resnet_v2_50 upload download_saved_model_from_path( 'https://storage.googleapis.com/inference-eu/models_zoo/' 'resnet_V2_50/saved_model/saved_model.pb') params['model_version'] = 2 upload_model(url, params, headers, UPLOAD_PART_SIZE) os.remove('saved_model.pb') # resnet model from tarball file_name = 'resnet_v2_fp16_savedmodel_NCHW.tar.gz' params = { 'model_name': MODEL_NAME + '-tar', 'model_version': 1, 'file_path': os.path.abspath(file_name), } download_saved_model_from_path( 'http://download.tensorflow.org/models/official' '/20181001_resnet/savedmodels' '/resnet_v2_fp16_savedmodel_NCHW.tar.gz', file_name) upload_model(url, params, headers, UPLOAD_PART_SIZE) os.remove(file_name) except Exception as e: pytest.fail(f"Unexpected error during upload test: {e.text}")
from management_api_tests.authenticate import get_user_token, get_admin_token, \ get_token MINIO_ACCESS_KEY_ID = os.environ.get('MINIO_ACCESS_KEY', 'AKIAIOSFODNN7EXAMPLE') MINIO_SECRET_ACCESS_KEY = os.environ.get( 'MINIO_SECRET_KEY', 'wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY') MINIO_ENDPOINT_ADDR = os.environ.get('MINIO_ENDPOINT_ADDR', 'http://127.0.0.1:9000') MINIO_REGION = os.environ.get('MINIO_REGION', 'us-east-1') SIGNATURE_VERSION = 's3v4' PORTABLE_SECRETS_PATHS = ['mgt-api/minio-access-info', 'mgt-api/tls-secret'] DEFAULT_HEADERS = { 'accept': 'application/json', 'Authorization': get_user_token()['id_token'], 'Content-Type': 'application/json', } ADMIN_HEADERS = { 'accept': 'application/json', 'Authorization': get_admin_token()['id_token'], 'Content-Type': 'application/json', } USER1_HEADERS = { 'accept': 'application/json', 'Authorization': get_token("mars")['id_token'], 'Content-Type': 'application/json', }