Python SageMakerHook.check_training_config Examples

Example #1

File: test_sagemaker_hook.py Project: zw39125432/airflow

    def test_check_valid_training(self, mock_check_url, mock_client):
        mock_client.return_value = None
        hook = SageMakerHook()
        hook.check_training_config(create_training_params)
        mock_check_url.assert_called_once_with(data_url)

        # InputDataConfig is optional, verify if check succeeds without InputDataConfig
        create_training_params_no_inputdataconfig = create_training_params.copy()
        create_training_params_no_inputdataconfig.pop("InputDataConfig")
        hook.check_training_config(create_training_params_no_inputdataconfig)

Example #2

 def test_check_valid_training(self, mock_check_url, mock_client):
     mock_client.return_value = None
     hook = SageMakerHook()
     hook.check_training_config(create_training_params)
     mock_check_url.assert_called_once_with(data_url)

Example #3

File: test_sagemaker_hook.py Project: nbrgil/incubator-airflow

 def test_check_valid_training(self, mock_check_url, mock_client):
     mock_client.return_value = None
     hook = SageMakerHook()
     hook.check_training_config(create_training_params)
     mock_check_url.assert_called_once_with(data_url)