From 12ad4a5c5c91f8b0b29e2f2fea08835279de34bc Mon Sep 17 00:00:00 2001
From: prcvih <41447049+autoih@users.noreply.github.com>
Date: Thu, 3 Oct 2019 09:02:22 -0700
Subject: [PATCH] fix ROC fit_predict arguments (#111)

---
 .../algorithms/postprocessing/reject_option_classification.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/aif360/algorithms/postprocessing/reject_option_classification.py b/aif360/algorithms/postprocessing/reject_option_classification.py
index e5ae8017..8cb8a68c 100644
--- a/aif360/algorithms/postprocessing/reject_option_classification.py
+++ b/aif360/algorithms/postprocessing/reject_option_classification.py
@@ -211,9 +211,9 @@ def predict(self, dataset):
 
         return dataset_new
 
-    def fit_predict(self, dataset):
+    def fit_predict(self, dataset_true, dataset_pred):
         """fit and predict methods sequentially."""
-        return self.fit().predict(dataset)
+        return self.fit(dataset_true, dataset_pred).predict(dataset_pred)
 
 # Function to obtain the pareto frontier
 def _get_pareto_frontier(costs, return_mask = True):  # <- Fastest for many points