@@ -782,19 +782,17 @@ def from_processing_name(cls, sagemaker_session, processing_job_name):
782
782
783
783
inputs = None
784
784
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" )),
790
789
dataset_definition = DatasetDefinition .from_boto (
791
- processing_input_dict .get ("DatasetDefinition" )
790
+ processing_input .get ("DatasetDefinition" )
792
791
),
792
+ app_managed = processing_input .get ("AppManaged" ) or False ,
793
793
)
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
+ ]
798
796
799
797
outputs = None
800
798
if job_desc .get ("ProcessingOutputConfig" ) and job_desc ["ProcessingOutputConfig" ].get (
@@ -804,16 +802,16 @@ def from_processing_name(cls, sagemaker_session, processing_job_name):
804
802
for processing_output_dict in job_desc ["ProcessingOutputConfig" ]["Outputs" ]:
805
803
processing_output = ProcessingOutput (
806
804
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
+ ),
807
809
)
810
+
808
811
if "S3Output" in processing_output_dict :
809
812
processing_output .source = processing_output_dict ["S3Output" ]["LocalPath" ]
810
813
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
+
817
815
outputs .append (processing_output )
818
816
output_kms_key = None
819
817
if job_desc .get ("ProcessingOutputConfig" ):
0 commit comments