Skip to content

Commit 24518f5

Browse files
committed
clean up code
1 parent 30c6dca commit 24518f5

File tree

1 file changed

+14
-16
lines changed

1 file changed

+14
-16
lines changed

src/sagemaker/processing.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -782,19 +782,17 @@ def from_processing_name(cls, sagemaker_session, processing_job_name):
782782

783783
inputs = None
784784
if job_desc.get("ProcessingInputs"):
785-
inputs = []
786-
for processing_input_dict in job_desc["ProcessingInputs"]:
787-
processing_input = ProcessingInput(
788-
input_name=processing_input_dict["InputName"],
789-
s3_input=S3Input.from_boto(processing_input_dict.get("S3Input")),
785+
inputs = [
786+
ProcessingInput(
787+
input_name=processing_input["InputName"],
788+
s3_input=S3Input.from_boto(processing_input.get("S3Input")),
790789
dataset_definition=DatasetDefinition.from_boto(
791-
processing_input_dict.get("DatasetDefinition")
790+
processing_input.get("DatasetDefinition")
792791
),
792+
app_managed=processing_input.get("AppManaged") or False,
793793
)
794-
if "AppManaged" in processing_input_dict:
795-
processing_input.app_managed = processing_input_dict["AppManaged"]
796-
797-
inputs.append(processing_input)
794+
for processing_input in job_desc["ProcessingInputs"]
795+
]
798796

799797
outputs = None
800798
if job_desc.get("ProcessingOutputConfig") and job_desc["ProcessingOutputConfig"].get(
@@ -804,16 +802,16 @@ def from_processing_name(cls, sagemaker_session, processing_job_name):
804802
for processing_output_dict in job_desc["ProcessingOutputConfig"]["Outputs"]:
805803
processing_output = ProcessingOutput(
806804
output_name=processing_output_dict["OutputName"],
805+
app_managed=processing_output_dict.get("AppManaged") or False,
806+
feature_store_output=FeatureStoreOutput.from_boto(
807+
processing_output_dict.get("FeatureStoreOutput")
808+
),
807809
)
810+
808811
if "S3Output" in processing_output_dict:
809812
processing_output.source = processing_output_dict["S3Output"]["LocalPath"]
810813
processing_output.destination = processing_output_dict["S3Output"]["S3Uri"]
811-
if "FeatureStoreOutput" in processing_output_dict:
812-
processing_output.feature_store_output = FeatureStoreOutput.from_boto(
813-
processing_output_dict["FeatureStoreOutput"]
814-
)
815-
if "AppManaged" in processing_output_dict:
816-
processing_output.app_managed = processing_output_dict["AppManaged"]
814+
817815
outputs.append(processing_output)
818816
output_kms_key = None
819817
if job_desc.get("ProcessingOutputConfig"):

0 commit comments

Comments
 (0)