Fix test

Author: pintaoz-aws

integ/test_codegen.py

@@ -42,7 +42,7 @@
 iris_df = iris_df[["target"] + [col for col in iris_df.columns if col != "target"]]
 train_data, test_data = train_test_split(iris_df, test_size=0.2, random_state=42)
 train_data.to_csv("./data/train.csv", index=False, header=False)
-test_data_no_target = test_data.drop('target', axis=1)
+test_data_no_target = test_data.drop("target", axis=1)
 
 # Upload Data
 prefix = "DEMO-scikit-iris"
@@ -150,11 +150,20 @@ def test_training_and_inference(self):
         )
         endpoint.wait_for_status("InService")
 
-        invoke_result = endpoint.invoke(body=test_data_no_target.to_csv(header=False, index=False),
-                                        content_type='text/csv',
-                                        accept='text/csv')
+        invoke_result = endpoint.invoke(
+            body=test_data_no_target.to_csv(header=False, index=False),
+            content_type="text/csv",
+            accept="text/csv",
+        )
+
+        assert invoke_result.body
 
-        print(invoke_result)
+        invoke_result = endpoint.invoke_with_response_stream(
+            test_data_no_target.to_csv(header=False, index=False),
+            content_type="text/csv",
+            accept="application/csv",
+        )
+        assert invoke_result.body.payload_part
 
     def test_intelligent_defaults(self):
         os.environ["SAGEMAKER_CORE_ADMIN_CONFIG_OVERRIDE"] = (

integ/test_experiment_and_trial.py

@@ -71,3 +71,8 @@ def test_experiment_and_trial(self):
         assert len(trial_component.parameters) == 7
         assert len(trial_component.metrics) == 1
         assert trial_component.metrics[0].count == 5
+
+        trial_component.disassociate_trail(trial_name=trial.trial_name)
+        trial_component.delete()
+        trial.delete()
+        experiment.delete()

src/sagemaker_core/main/code_injection/constants.py

@@ -13,7 +13,7 @@
 """Constants used in the code_injection modules."""
 from enum import Enum
 
-BASIC_TYPES = ["string", "boolean", "integer", "long", "double", "timestamp", "float"]
+BASIC_TYPES = ["string", "boolean", "integer", "long", "double", "timestamp", "float", "blob"]
 STRUCTURE_TYPE = "structure"
 MAP_TYPE = "map"
 LIST_TYPE = "list"

tst/generated/test_resources.py

@@ -378,8 +378,8 @@ def _convert_dict_keys_into_pascal_case(self, input_args: dict):
         return converted
 
     def _convert_to_pascal(self, string: str):
-        if string == "auto_ml_job_name":
-            return "AutoMLJobName"
+        if string.startswith("auto_ml_"):
+            return "AutoML" + snake_to_pascal(string[7:])
         return snake_to_pascal(string)
 
     def _get_required_parameters_for_function(self, func) -> dict: