Added json logging for python mllib tests