Update roc_metrics.py

Author: edosedgar

metrics/roc_metrics.py

@@ -3,7 +3,7 @@
 """
 import warnings
 warnings.simplefilter(action='ignore', category=FutureWarning)
-from sklearn.metrics import roc_curve, auc
+from sklearn.metrics import roc_curve, auc, roc_auc_score
 import numpy as np
 import torch.nn.functional as F
 import torch
@@ -118,4 +118,4 @@ def compute(self):
         """Computes PE based on inputs passed in to ``update`` previously."""
         x = dim_zero_cat(self.x)
         y = dim_zero_cat(self.y)
-        return pe(y, x)
+        return pe(y, x)