def augment(module_name, base_class):
for name, cls in inspect.getmembers(sys.modules[module_name],<EOL>lambda x : inspect.isclass(x) and  issubclass(x, base_class) ):<EOL><INDENT>if cls == base_class:<EOL><INDENT>continue<EOL><DEDENT>cls.augment()<EOL><DEDENT>
Call the augment() method for all of the derived classes in the module