diff --git a/test/test_distributed.py b/test/test_distributed.py index 09ab0fa4f..84cfff046 100644 --- a/test/test_distributed.py +++ b/test/test_distributed.py @@ -235,7 +235,7 @@ def test_elastic_training_dl2(self, backend) -> None: "--run_path", f"--nnodes={nnodes}", f"--nproc_per_node={world_size}", - abs_path("bin/elastic_training.py"), + abs_path("elastic_training.py"), "--" + backend, "--dl2", ], @@ -258,7 +258,7 @@ def test_elastic_training_dl1(self, backend) -> None: "--run_path", f"--nnodes={nnodes}", f"--nproc_per_node={world_size}", - abs_path("bin/elastic_training.py"), + abs_path("elastic_training.py"), "--" + backend, "--dl1", ],