Sagemaker test fix (#10987)

* wrong makefile command * ddp test fix
2021-03-31 13:44:22 +02:00
parent 645f45c462
commit ced7284a60
2 changed files with 3 additions and 3 deletions
--- a/tests/sagemaker/test_multi_node_data_parallel.py
+++ b/tests/sagemaker/test_multi_node_data_parallel.py
@@ -58,7 +58,7 @@ class MultiNodeTest(unittest.TestCase):
    def create_estimator(self, instance_count):
        job_name = f"{self.env.base_job_name}-{instance_count}-{'ddp' if 'ddp' in self.script else 'smd'}"
        # distributed data settings
-        distribution = {"smdistributed": {"dataparallel": {"enabled": True}}}
+        distribution = {"smdistributed": {"dataparallel": {"enabled": True}}} if self.script != "run_ddp.py" else None

        # creates estimator
        return HuggingFace(