fix: disable profiler in some release tests (#2261)

icywang86rui · Rui Wang Napieralski · web-flow · commit 2b2b4ddad153 · 2021-03-31T16:45:48.000-07:00
Co-authored-by: Rui Wang Napieralski &lt;wru@amazon.com&gt;
diff --git a/tests/integ/test_horovod.py b/tests/integ/test_horovod.py
@@ -91,6 +91,7 @@ def _create_and_fit_estimator(sagemaker_session, tf_version, py_version, instanc
         py_version=py_version,
         framework_version=tf_version,
         distribution={"mpi": {"enabled": True}},
+        disable_profiler=True,
     )
 
     with timeout.timeout(minutes=integ.TRAINING_DEFAULT_TIMEOUT_MINUTES):
diff --git a/tests/integ/test_huggingface.py b/tests/integ/test_huggingface.py
@@ -58,6 +58,7 @@ def test_huggingface_training(
                 "repo": "https://github.com/huggingface/transformers.git",
                 "branch": f"v{huggingface_training_latest_version}",
             },
+            disable_profiler=True,
         )
 
         train_input = hf.sagemaker_session.upload_data(
diff --git a/tests/integ/test_tf.py b/tests/integ/test_tf.py
@@ -141,6 +141,7 @@ def test_mnist_distributed(
         framework_version=tensorflow_training_latest_version,
         py_version=tensorflow_training_latest_py_version,
         distribution=PARAMETER_SERVER_DISTRIBUTION,
+        disable_profiler=True,
     )
     inputs = estimator.sagemaker_session.upload_data(
         path=os.path.join(MNIST_RESOURCE_PATH, "data"), key_prefix="scriptmode/distributed_mnist"

Original file line number	Diff line number	Diff line change
`@@ -91,6 +91,7 @@ def _create_and_fit_estimator(sagemaker_session, tf_version, py_version, instanc`
`91`	`91`	`py_version=py_version,`
`92`	`92`	`framework_version=tf_version,`
`93`	`93`	`distribution={"mpi": {"enabled": True}},`
	`94`	`+ disable_profiler=True,`
`94`	`95`	`)`
`95`	`96`
`96`	`97`	`with timeout.timeout(minutes=integ.TRAINING_DEFAULT_TIMEOUT_MINUTES):`
Original file line number	Diff line number	Diff line change
`@@ -58,6 +58,7 @@ def test_huggingface_training(`
`58`	`58`	`"repo": "https://github.com/huggingface/transformers.git",`
`59`	`59`	`"branch": f"v{huggingface_training_latest_version}",`
`60`	`60`	`},`
	`61`	`+ disable_profiler=True,`
`61`	`62`	`)`
`62`	`63`
`63`	`64`	`train_input = hf.sagemaker_session.upload_data(`
Original file line number	Diff line number	Diff line change
`@@ -141,6 +141,7 @@ def test_mnist_distributed(`
`141`	`141`	`framework_version=tensorflow_training_latest_version,`
`142`	`142`	`py_version=tensorflow_training_latest_py_version,`
`143`	`143`	`distribution=PARAMETER_SERVER_DISTRIBUTION,`
	`144`	`+ disable_profiler=True,`
`144`	`145`	`)`
`145`	`146`	`inputs = estimator.sagemaker_session.upload_data(`
`146`	`147`	`path=os.path.join(MNIST_RESOURCE_PATH, "data"), key_prefix="scriptmode/distributed_mnist"`