I'd suggest making this an optional argument to DenseDesignMatrix.__init__ that MNIST then passes to the super __init__