class SimpleModule:

    def __init__(self,
                 in_features: int,
                 out_features: int,
                 activation: str = "relu"):
        self.in_features = in_features
        self.out_features = out_features
        self.activation = activation
        print(
            f"SimpleModule created: {in_features} -> {out_features}, activation: {activation}"
        )