jonescompneurolab · cjayb · May 6, 2021 · Apr 29, 2021 · Apr 29, 2021 · Apr 29, 2021
diff --git a/doc/api.rst b/doc/api.rst
@@ -14,12 +14,9 @@ Simulation (:py:mod:`hnn_core`):
 .. autosummary::
    :toctree: generated/
 
-   L2Pyr
-   L5Pyr
-   L2Basket
-   L5Basket
    simulate_dipole
    Network
+   Cell
    CellResponse
 
 Dipole (:py:mod:`hnn_core.dipole`):

diff --git a/hnn_core/__init__.py b/hnn_core/__init__.py
@@ -3,8 +3,7 @@
 from .params import Params, read_params
 from .network import Network
 from .cell_response import CellResponse, read_spikes
-from .pyramidal import L2Pyr, L5Pyr
-from .basket import L2Basket, L5Basket
+from .cells_default import pyramidal, basket
 from .parallel_backends import MPIBackend, JoblibBackend
 
 __version__ = '0.2.dev0'
diff --git a/hnn_core/basket.py b/hnn_core/basket.py