old_contents: from sklearn.externals import joblib
  
  class SkMnist(object):
      def __init__(self):
          self.class_names = ["class:{}".format(str(i)) for i in range(10)]
          self.clf = joblib.load('/data/sk.pkl') 
  
      def predict(self,X,feature_names):
          predictions = self.clf.predict(X)
          return predictions
  
      
  
new_contents: from sklearn.externals import joblib
  
  class SkMnist(object):
      def __init__(self):
          self.class_names = ["class:{}".format(str(i)) for i in range(10)]
          self.clf = joblib.load('/data/sk.pkl') 
  
      def predict(self,X,feature_names):
          predictions = self.clf.predict_proba(X)
          return predictions
  
      
  
subject: change predict to predict_proba in sk runtime