Remove try catch so error reflected if s3uri property not found and r…

…emoved unused imports

src/stepfunctions/steps/sagemaker.py

@@ -118,14 +118,8 @@ def __init__(self, state_id, estimator, job_name, data=None, hyperparameters=Non
  if data is not None and is_data_placeholder:
  # Replace the 'S3Uri' key with one that supports JSONpath value.
  # Support for uri str only: The list will only contain 1 element
- try:
- data_uri = parameters['InputDataConfig'][0]['DataSource']['S3DataSource'].pop('S3Uri', None)
- parameters['InputDataConfig'][0]['DataSource']['S3DataSource']['S3Uri.$'] = data_uri
- except KeyError as error:
- logger.warning(f"Placeholder {data} used for data, but could not locate S3Uri property to make "
- f"placeholder compatible.\n"
- f"Path used:['InputDataConfig'][0]['DataSource']['S3DataSource']['S3Uri'] - "
- f"{error} key was not found.")
+ data_uri = parameters['InputDataConfig'][0]['DataSource']['S3DataSource'].pop('S3Uri', None)
+ parameters['InputDataConfig'][0]['DataSource']['S3DataSource']['S3Uri.$'] = data_uri
 
  if hyperparameters is not None:
  parameters['HyperParameters'] = hyperparameters

tests/integ/test_sagemaker_steps.py

@@ -29,7 +29,6 @@
 from sagemaker.tuner import HyperparameterTuner
 from sagemaker.processing import ProcessingInput, ProcessingOutput
 
-from stepfunctions.inputs import ExecutionInput, StepInput
 from stepfunctions.steps import Chain
 from stepfunctions.steps.sagemaker import TrainingStep, TransformStep, ModelStep, EndpointStep, EndpointConfigStep, TuningStep, ProcessingStep
 from stepfunctions.workflow import Workflow