diff --git a/tests/integ/test_processing.py b/tests/integ/test_processing.py index 8ceb3f2195..3be778ba84 100644 --- a/tests/integ/test_processing.py +++ b/tests/integ/test_processing.py @@ -20,7 +20,6 @@ from sagemaker import image_uris, Session from sagemaker.dataset_definition.inputs import ( DatasetDefinition, - RedshiftDatasetDefinition, AthenaDatasetDefinition, S3Input, ) @@ -766,25 +765,6 @@ def _get_processing_inputs_with_all_parameters(bucket): s3_compression_type="None", ), ), - ProcessingInput( - input_name="redshift_dataset_definition", - app_managed=True, - dataset_definition=DatasetDefinition( - local_path="/opt/ml/processing/input/rdd", - data_distribution_type="FullyReplicated", - input_mode="File", - redshift_dataset_definition=RedshiftDatasetDefinition( - cluster_id="integ-test-cluster-prod-us-west-2", - database="dev", - db_user="awsuser", - query_string="SELECT * FROM shoes", - cluster_role_arn="arn:aws:iam::037210630505:role/RedshiftClusterRole-prod-us-west-2", - output_s3_uri=f"s3://{bucket}/rdd", - output_format="CSV", - output_compression="None", - ), - ), - ), ProcessingInput( input_name="athena_dataset_definition", app_managed=True, @@ -853,25 +833,6 @@ def _get_processing_job_inputs_and_outputs(bucket, output_kms_key): "S3CompressionType": "None", }, }, - { - "InputName": "redshift_dataset_definition", - "AppManaged": True, - "DatasetDefinition": { - "RedshiftDatasetDefinition": { - "ClusterId": "integ-test-cluster-prod-us-west-2", - "Database": "dev", - "DbUser": "awsuser", - "QueryString": "SELECT * FROM shoes", - "ClusterRoleArn": "arn:aws:iam::037210630505:role/RedshiftClusterRole-prod-us-west-2", - "OutputS3Uri": f"s3://{bucket}/rdd", - "OutputFormat": "CSV", - "OutputCompression": "None", - }, - "LocalPath": "/opt/ml/processing/input/rdd", - "DataDistributionType": "FullyReplicated", - "InputMode": "File", - }, - }, { "InputName": "athena_dataset_definition", "AppManaged": True,