The docstring for input_space also needs to be updated, since you don't want to specify it if this MLP is a sub-MLP.