From ac68eff9fa8c723730039992963bb5e001a43446 Mon Sep 17 00:00:00 2001 From: Jirka Borovec Date: Mon, 5 Jul 2021 12:36:16 +0200 Subject: [PATCH] Apply suggestions from code review --- torchmetrics/classification/kl_divergence.py | 2 +- torchmetrics/classification/stat_scores.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/torchmetrics/classification/kl_divergence.py b/torchmetrics/classification/kl_divergence.py index c8935d60ae2..8fe9bd64446 100644 --- a/torchmetrics/classification/kl_divergence.py +++ b/torchmetrics/classification/kl_divergence.py @@ -60,7 +60,7 @@ class KLDivergence(Metric): >>> kldivergence(p, q) tensor(0.0853) """ - # canot be used because if scripting + # TODO: canot be used because if scripting # measures: Union[List[Tensor], Tensor] total: Tensor diff --git a/torchmetrics/classification/stat_scores.py b/torchmetrics/classification/stat_scores.py index 87dab7927ca..14ae961fbe1 100644 --- a/torchmetrics/classification/stat_scores.py +++ b/torchmetrics/classification/stat_scores.py @@ -130,7 +130,7 @@ class StatScores(Metric): """ - # canot be used because if scripting + # TODO: canot be used because if scripting # tp: Union[Tensor, List[Tensor]] # fp: Union[Tensor, List[Tensor]] # tn: Union[Tensor, List[Tensor]]