Skip to content

Commit

Permalink
Fix incorrect function name called in unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
deadlycoconuts committed Nov 7, 2024
1 parent 163f60b commit 76ecdc5
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions python/sdk/test/utils_unit_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand All @@ -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
Expand Down

0 comments on commit 76ecdc5

Please sign in to comment.