def run_for_estimator(self, estimator, test_set): X_test, y_test = test_set dist = estimator.predict(X_test) assert len(dist) == 2 err = DE.rms(y_test, dist) return err