1 files changed, 6 insertions, 0 deletions

diff --git a/gen/__init__.py b/gen/__init__.py
index e69de29..8c60b46 100644
--- a/gen/__init__.py
+++ b/gen/__init__.py
@@ -0,0 +1,6 @@
+import generator as gen
+import edf_generators as edf
+
+gen.register_generator("G-EDF", edf.GedfGenerator)
+gen.register_generator("P-EDF", edf.PedfGenerator)
+gen.register_generator("C-EDF", edf.CedfGenerator)