From 5fc805ee25c546c710b95b05543397f76e1d714e Mon Sep 17 00:00:00 2001 From: ifielker Date: Thu, 12 Mar 2020 15:24:50 -0400 Subject: [PATCH] Firebase ML Changing service endpoint --- firebase_admin/ml.py | 4 ++-- tests/test_ml.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/firebase_admin/ml.py b/firebase_admin/ml.py index b7d8b818b..427ff6bdd 100644 --- a/firebase_admin/ml.py +++ b/firebase_admin/ml.py @@ -773,8 +773,8 @@ def _validate_page_token(page_token): class _MLService: """Firebase ML service.""" - PROJECT_URL = 'https://mlkit.googleapis.com/v1beta1/projects/{0}/' - OPERATION_URL = 'https://mlkit.googleapis.com/v1beta1/' + PROJECT_URL = 'https://firebaseml.googleapis.com/v1beta2/projects/{0}/' + OPERATION_URL = 'https://firebaseml.googleapis.com/v1beta2/' POLL_EXPONENTIAL_BACKOFF_FACTOR = 1.5 POLL_BASE_WAIT_TIME_SECONDS = 3 diff --git a/tests/test_ml.py b/tests/test_ml.py index 6accba1cb..fb85c9d4e 100644 --- a/tests/test_ml.py +++ b/tests/test_ml.py @@ -24,7 +24,7 @@ from tests import testutils -BASE_URL = 'https://mlkit.googleapis.com/v1beta1/' +BASE_URL = 'https://firebaseml.googleapis.com/v1beta2/' PROJECT_ID = 'myProject1' PAGE_TOKEN = 'pageToken' NEXT_PAGE_TOKEN = 'nextPageToken' @@ -306,7 +306,7 @@ def instrument_ml_service(status=200, payload=None, operations=False, app=None): app = firebase_admin.get_app() ml_service = ml._get_ml_service(app) recorder = [] - session_url = 'https://mlkit.googleapis.com/v1beta1/' + session_url = 'https://firebaseml.googleapis.com/v1beta2/' if isinstance(status, list): adapter = testutils.MockMultiRequestAdapter