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("MC2", edf.Mc2Generator) #gen.register_generator("C-EDF", edf.CedfGenerator)