Fix namespace issues

examples/snapshot.py

@@ -4,7 +4,7 @@
 """
 import numpy as np
 
-import pycola
+import pycola3
 
 # Cosmological parameters
 omega_m = 0.316
@@ -40,7 +40,7 @@ def particle_mass(z):
 print("Min. halo mass: %3.2e Msun" % halo_mmin)
 
 # Calculate ZA initial conditions
-sx, sy, sz = pycola.ic.ic_za(
+sx, sy, sz = pycola3.ic.ic_za(
     "camb_matterpower_z0.dat", boxsize=BOXSIZE, npart=NPART, init_seed=SEED
 )
 cellsize = BOXSIZE / float(sx.shape[0])
@@ -49,10 +49,10 @@ def particle_mass(z):
 print("gridcellsize:", gridcellsize)
 
 # Calculate 2LPT positions
-sx2, sy2, sz2 = pycola.ic.ic_2lpt(cellsize, sx, sy, sz, boxsize=BOXSIZE)
+sx2, sy2, sz2 = pycola3.ic.ic_2lpt(cellsize, sx, sy, sz, boxsize=BOXSIZE)
 
 # Evolve particles
-px, py, pz, vx, vy, vz, *_ = pycola.evolve.evolve(
+px, py, pz, vx, vy, vz, *_ = pycola3.evolve.evolve(
     cellsize,
     sx,
     sy,

pycola3/__init__.py

@@ -1 +1,2 @@
-from . import acceleration, aux, box_smooth, cic, evolve, growth, ic, potential
+# from . import acceleration, aux, box_smooth, cic, evolve, growth, ic, potential
+from . import aux, evolve, growth, ic

setup.py

@@ -26,10 +26,10 @@
 from setuptools import Extension, find_namespace_packages, setup
 
 sources = {
-    "pycola3._cic": "pycola3/src/cic.pyx",
-    "pycola3._potential": "pycola3/src/potential.pyx",
-    "pycola3._acceleration": "pycola3/src/acceleration.pyx",
-    "pycola3._box_smooth": "pycola3/src/box_smooth.pyx",
+    "pycola3.cic": "pycola3/src/cic.pyx",
+    "pycola3.potential": "pycola3/src/potential.pyx",
+    "pycola3.acceleration": "pycola3/src/acceleration.pyx",
+    "pycola3.box_smooth": "pycola3/src/box_smooth.pyx",
 }
 
 # Compile and link arguments for Cython modules
@@ -101,7 +101,7 @@
 
 setup_args = {
     "name": "pycola3",
-    "version": "0.0.2",
+    "version": "0.0.3",
     "author": "Phil Bull / Svetlin V. Tassev",
     "author_email": "[email protected]",
     "description": "A Python/Cython N-body code, implementing the Comoving Lagrangian Acceleration (COLA) method in the temporal and spatial domains.",