Fixing imports for mesos

galaxyproject · Apr 21, 2015 · fe3e919 · fe3e919
1 parent d3287fa
commit fe3e919
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pulsar/mesos/__init__.py b/pulsar/mesos/__init__.py
@@ -1,19 +1,21 @@
 """ This module and submodules contain code for interfacing the Apache Mesos framework.
 """
 try:
-    from mesos import (
+    from mesos.interface import (
         Scheduler,
-        Executor,
+        Executor
+    )
+    from mesos.native import (
         MesosSchedulerDriver,
-        MesosExecutorDriver,
+        MesosExecutorDriver
     )
 except ImportError:
     Scheduler = object
     Executor = object
     MesosSchedulerDriver = None
     MesosExecutorDriver = None
 try:
-    import mesos_pb2
+    from mesos.interface import mesos_pb2
 except ImportError:
     mesos_pb2 = None