Update test_results.py

2021-03-24 01:57:31 -07:00 · 2021-03-24 01:57:31 -07:00 · 9d4a2b891d
parent c5053da789
commit 9d4a2b891d
1 changed files with 1 additions and 0 deletions
--- a/tests/core/test_results.py
+++ b/tests/core/test_results.py
@ -30,6 +30,7 @@ from tests.helpers.runif import RunIf

 def _setup_ddp(rank, worldsize):
    import os
+    
    os.environ["MASTER_ADDR"] = "localhost"

    # initialize the process group