Updated test_workflow and est_training_step

Author: Marius Moisescu

tests/integ/sagemaker/workflow/test_workflow.py

@@ -1269,8 +1269,6 @@ def test_caching_behavior(
         # create pipeline
         pipeline.create(role)
         definition = json.loads(pipeline.definition())
-        # delete profiler config for assertions as it will contain a timestamp
-        del definition["Steps"][1]["Arguments"]["ProfilerRuleConfigurations"]
 
         # verify input path
         expected_abalone_input_path = f"{pipeline_name}/{step_process.name}" f"/input/abalone_data"

tests/unit/sagemaker/workflow/test_training_step.py

@@ -532,9 +532,6 @@ def test_training_step_with_framework_estimator(
     del expected_step_args["OutputDataConfig"]["S3OutputPath"]
     del step_def["Arguments"]["OutputDataConfig"]["S3OutputPath"]
 
-    # delete profiler rule configurations because of timestamp collision
-    del expected_step_args["ProfilerRuleConfigurations"]
-
     if "sagemaker_s3_output" in step_args["HyperParameters"]:
         del expected_step_args["HyperParameters"]["sagemaker_s3_output"]
         del step_def["Arguments"]["HyperParameters"]["sagemaker_s3_output"]
@@ -602,9 +599,6 @@ def test_training_step_with_framework_estimator_local_code(
     del expected_step_args["OutputDataConfig"]["S3OutputPath"]
     del step_def["Arguments"]["OutputDataConfig"]["S3OutputPath"]
 
-    # delete profiler rule configurations because of timestamp collision
-    del expected_step_args["ProfilerRuleConfigurations"]
-
     if "sagemaker_s3_output" in step_args["HyperParameters"]:
         del expected_step_args["HyperParameters"]["sagemaker_s3_output"]
         del step_def["Arguments"]["HyperParameters"]["sagemaker_s3_output"]
@@ -694,9 +688,6 @@ def test_training_step_with_algorithm_base(algo_estimator, training_input, pipel
     del step_args["InputDataConfig"][0]["DataSource"]["S3DataSource"]["S3Uri"]
     del step_def["Arguments"]["InputDataConfig"][0]["DataSource"]["S3DataSource"]["S3Uri"]
 
-    # delete profiler rule configurations because of timestamp collision
-    del step_args["ProfilerRuleConfigurations"]
-
     assert step_def == {
         "Name": "MyTrainingStep",
         "Type": "Training",
@@ -781,9 +772,6 @@ def test_training_step_with_algorithm_base_local_code(
     del step_args["InputDataConfig"][0]["DataSource"]["S3DataSource"]["S3Uri"]
     del step_def["Arguments"]["InputDataConfig"][0]["DataSource"]["S3DataSource"]["S3Uri"]
 
-    # delete profiler rule configurations because of timestamp collision
-    del step_args["ProfilerRuleConfigurations"]
-
     assert step_def == {
         "Name": "MyTrainingStep",
         "Type": "Training",