Apppublic

Underground-Digital/Workflow-Engine

sourceHugging Faceupdated 2y agoView on Hugging Face

0likes

rootapitestsintegration_testsmodel_runtimesagemakertest_provider.py

test_provider.py14 linesDownload Raw Back to sagemaker

1import pytest2 3from core.model_runtime.errors.validate import CredentialsValidateFailedError4from core.model_runtime.model_providers.sagemaker.sagemaker import SageMakerProvider5 6 7def test_validate_provider_credentials():8    provider = SageMakerProvider()9 10    with pytest.raises(CredentialsValidateFailedError):11        provider.validate_provider_credentials(credentials={})12 13    provider.validate_provider_credentials(credentials={})14