Literal

Co-authored-by: Daniel Stancl <[email protected]>
Lightning-AI · Dec 22, 2023 · ef7da19 · ef7da19
1 parent 0546c9a
commit ef7da19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchmetrics/retrieval/auroc.py b/src/torchmetrics/retrieval/auroc.py
@@ -97,7 +97,7 @@ class RetrievalAUROC(RetrievalMetric):
 
     def __init__(
         self,
-        empty_target_action: str = "neg",
+        empty_target_action: Literal["error",  "skip", "neg", "pos"] = "neg",
         ignore_index: Optional[int] = None,
         top_k: Optional[int] = None,
         max_fpr: Optional[float] = None,