Make users import xactor instead xactor.mpi_actor

parantapa · Dec 4, 2019 · 42a9979 · 42a9979
1 parent 5c073d5
commit 42a9979
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 3 deletions.
diff --git a/test_xactor/test_fanout.py b/test_xactor/test_fanout.py
@@ -4,7 +4,7 @@
 
 import random
 import logging
-import xactor.mpi_actor as xa
+import xactor as xa
 
 MIN_NUM = 100
 MAX_NUM = 200

diff --git a/test_xactor/test_hello.py b/test_xactor/test_hello.py
@@ -1,7 +1,7 @@
 """Hello World in XActor."""
 
 import logging
-import xactor.mpi_actor as xa
+import xactor as xa
 
 class Greeter:
     def greet(self, name):

diff --git a/test_xactor/test_ping_pong.py b/test_xactor/test_ping_pong.py
@@ -3,7 +3,7 @@
 from time import time
 
 import logging
-import xactor.mpi_actor as xa
+import xactor as xa
 
 class Worker:
     def __init__(self):

diff --git a/xactor/__init__.py b/xactor/__init__.py
@@ -1 +1,3 @@
+"""XActor: A Distributed Actor Programming Framework for Python."""
 
+from xactor.mpi_actor import *
diff --git a/xactor/mpi_actor.py b/xactor/mpi_actor.py
@@ -18,6 +18,7 @@
     "delete_actors",
     "MASTER_RANK",
     "EVERY_RANK",
+    "WORLD_SIZE",
     "getLogger",
 ]