each function class in separate source file

1 files changed, 6 insertions, 3 deletions

diff --git a/pyspike/__init__.py b/pyspike/__init__.py
index 55687e6..f480964 100644
--- a/pyspike/__init__.py
+++ b/pyspike/__init__.py
@@ -4,10 +4,13 @@ Copyright 2014, Mario Mulansky <mario.mulansky@gmx.net>
 Distributed under the BSD License
 """
 
-__all__ = ["function", "distances", "spikes"]
+__all__ = ["distances", "spikes", "PieceWiseConstFunc", "PieceWiseLinFunc",
+           "DiscreteFunc"]
+
+from PieceWiseConstFunc import PieceWiseConstFunc
+from PieceWiseLinFunc import PieceWiseLinFunc
+from DiscreteFunc import DiscreteFunc
 
-from function import PieceWiseConstFunc, PieceWiseLinFunc, \
-    DiscreteFunction, average_profile
 from distances import isi_profile, isi_distance, \
     spike_profile, spike_distance, \
     spike_sync_profile, spike_sync, \