the lab imports of source.mesh and source.catalog directly

bccp · Apr 4, 2017 · e3e6e13 · e3e6e13
1 parent 3e9fbed
commit e3e6e13
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/nbodykit/lab.py b/nbodykit/lab.py
@@ -1,11 +1,15 @@
 from mpi4py import MPI
-
 import numpy
 
-from nbodykit.batch import TaskManager 
-from nbodykit import source as Source, cosmology
+# sources
+from nbodykit.source.catalog import *
+from nbodykit.source.mesh import *
+
+# algorithms
 from nbodykit.algorithms import *
+
+from nbodykit.batch import TaskManager 
+from nbodykit import cosmology
 from nbodykit import CurrentMPIComm
 from nbodykit import transform
-
 from nbodykit import io as IO