diff --git a/python/xoscar/collective/gloo/test/test_pygloo.py b/python/xoscar/collective/gloo/test/test_pygloo.py index 2c4b0573..ca1bf5d7 100644 --- a/python/xoscar/collective/gloo/test/test_pygloo.py +++ b/python/xoscar/collective/gloo/test/test_pygloo.py @@ -10,6 +10,7 @@ fileStore_path = "./collective" system_name = platform.system() + def worker_allgather(rank): from .. import xoscar_pygloo as xp diff --git a/python/xoscar/collective/test/test_pygloo_tcp_store.py b/python/xoscar/collective/test/test_pygloo_tcp_store.py index e285fb51..f72a88b9 100644 --- a/python/xoscar/collective/test/test_pygloo_tcp_store.py +++ b/python/xoscar/collective/test/test_pygloo_tcp_store.py @@ -11,6 +11,7 @@ fileStore_path = "./collective" system_name = platform.system() + def worker_allgather(rank): from ..gloo import xoscar_pygloo as xp from ..rendezvous import xoscar_store as xs