From 76ecdc5315c50773ef6bd6a781bc1061c2ceffa0 Mon Sep 17 00:00:00 2001 From: ewezy Date: Thu, 7 Nov 2024 15:08:48 +0800 Subject: [PATCH] Fix incorrect function name called in unit test --- python/sdk/test/utils_unit_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sdk/test/utils_unit_test.py b/python/sdk/test/utils_unit_test.py index 9763b8c67..712bbc272 100644 --- a/python/sdk/test/utils_unit_test.py +++ b/python/sdk/test/utils_unit_test.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from merlin.util import guess_mlp_ui_url, valid_name_check, get_bucket_name, get_artifact_path +from merlin.util import guess_mlp_ui_url, valid_name_check, get_blob_storage_scheme, get_bucket_name, get_artifact_path import pytest @@ -37,10 +37,10 @@ def test_name_check(): @pytest.mark.unit def test_get_blob_storage_scheme(): gcs_artifact_uri = 'gs://some-bucket/mlflow/81/ddd' - assert test_get_blob_storage_scheme(gcs_artifact_uri) == 'gs' + assert get_blob_storage_scheme(gcs_artifact_uri) == 'gs' s3_artifact_uri = 's3://some-bucket/mlflow/81/ddd' - assert test_get_blob_storage_scheme(s3_artifact_uri) == 's3' + assert get_blob_storage_scheme(s3_artifact_uri) == 's3' @pytest.mark.unit