patch Softmax pickles in __setstate__
PEP8