Please use `super(TextAnnotationModule, self).__init__(*args, **kwargs)` to get the best behavior.