Added test_pytorch_compiler.py

Marius Moisescu · Marius Moisescu · commit 8523702ae415 · 2022-12-09T22:51:38.000Z
diff --git a/tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py b/tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py
@@ -144,7 +144,10 @@ def _create_train_job(version, instance_type, training_compiler_config, instance
                 "RuleParameters": {"rule_to_invoke": "ProfilerReport"},
             }
         ],
-        "profiler_config": {"S3OutputPath": "s3://{}/".format(BUCKET_NAME)},
+        "profiler_config": {
+            "DisableProfiler": False,
+            "S3OutputPath": "s3://{}/".format(BUCKET_NAME)
+        },
     }
 
 

Original file line number	Diff line number	Diff line change
`@@ -144,7 +144,10 @@ def _create_train_job(version, instance_type, training_compiler_config, instance`
`144`	`144`	`"RuleParameters": {"rule_to_invoke": "ProfilerReport"},`
`145`	`145`	`}`
`146`	`146`	`],`
`147`		`- "profiler_config": {"S3OutputPath": "s3://{}/".format(BUCKET_NAME)},`
	`147`	`+ "profiler_config": {`
	`148`	`+ "DisableProfiler": False,`
	`149`	`+ "S3OutputPath": "s3://{}/".format(BUCKET_NAME)`
	`150`	`+ },`
`148`	`151`	`}`
`149`	`152`
`150`	`153`