From 6652c5523421af9c534c3321d220e02c1941c95c Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 13 Dec 2022 14:50:25 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/torchmetrics/__init__.py | 26 ++++++------ src/torchmetrics/audio/__init__.py | 10 ++--- src/torchmetrics/detection/__init__.py | 2 +- src/torchmetrics/detection/mean_ap.py | 2 +- src/torchmetrics/functional/__init__.py | 4 +- src/torchmetrics/functional/audio/__init__.py | 10 ++--- .../functional/classification/__init__.py | 40 +++++++++---------- src/torchmetrics/functional/image/__init__.py | 16 ++++---- .../functional/multimodal/__init__.py | 2 +- .../functional/nominal/__init__.py | 8 ++-- .../functional/pairwise/__init__.py | 8 ++-- .../functional/regression/__init__.py | 30 +++++++------- .../functional/retrieval/__init__.py | 18 ++++----- src/torchmetrics/functional/text/__init__.py | 30 +++++++------- src/torchmetrics/image/__init__.py | 22 +++++----- src/torchmetrics/multimodal/__init__.py | 2 +- src/torchmetrics/nominal/__init__.py | 8 ++-- src/torchmetrics/regression/__init__.py | 32 +++++++-------- src/torchmetrics/retrieval/__init__.py | 20 +++++----- src/torchmetrics/text/__init__.py | 30 +++++++------- src/torchmetrics/utilities/__init__.py | 8 ++-- src/torchmetrics/utilities/imports.py | 2 +- src/torchmetrics/wrappers/__init__.py | 10 ++--- tests/unittests/__init__.py | 2 +- 24 files changed, 171 insertions(+), 171 deletions(-) diff --git a/src/torchmetrics/__init__.py b/src/torchmetrics/__init__.py index 235a7078f39..96426e39cd4 100644 --- a/src/torchmetrics/__init__.py +++ b/src/torchmetrics/__init__.py @@ -2,7 +2,7 @@ import logging as __logging import os -from torchmetrics.__about__ import * # noqa: F401, F403 +from torchmetrics.__about__ import * _logger = __logging.getLogger("torchmetrics") _logger.addHandler(__logging.StreamHandler()) @@ -11,16 +11,16 @@ _PACKAGE_ROOT = os.path.dirname(__file__) _PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT) -from torchmetrics import functional # noqa: E402 -from torchmetrics.aggregation import CatMetric, MaxMetric, MeanMetric, MinMetric, SumMetric # noqa: E402 -from torchmetrics.audio import ( # noqa: E402 +from torchmetrics import functional +from torchmetrics.aggregation import CatMetric, MaxMetric, MeanMetric, MinMetric, SumMetric +from torchmetrics.audio import ( PermutationInvariantTraining, ScaleInvariantSignalDistortionRatio, ScaleInvariantSignalNoiseRatio, SignalDistortionRatio, SignalNoiseRatio, ) -from torchmetrics.classification import ( # noqa: E402 +from torchmetrics.classification import ( AUROC, ROC, Accuracy, @@ -42,8 +42,8 @@ Specificity, StatScores, ) -from torchmetrics.collections import MetricCollection # noqa: E402 -from torchmetrics.image import ( # noqa: E402 +from torchmetrics.collections import MetricCollection +from torchmetrics.image import ( ErrorRelativeGlobalDimensionlessSynthesis, MultiScaleStructuralSimilarityIndexMeasure, PeakSignalNoiseRatio, @@ -53,9 +53,9 @@ TotalVariation, UniversalImageQualityIndex, ) -from torchmetrics.metric import Metric # noqa: E402 -from torchmetrics.nominal import CramersV, PearsonsContingencyCoefficient, TheilsU, TschuprowsT # noqa: E402 -from torchmetrics.regression import ( # noqa: E402 +from torchmetrics.metric import Metric +from torchmetrics.nominal import CramersV, PearsonsContingencyCoefficient, TheilsU, TschuprowsT +from torchmetrics.regression import ( ConcordanceCorrCoef, CosineSimilarity, ExplainedVariance, @@ -73,7 +73,7 @@ TweedieDevianceScore, WeightedMeanAbsolutePercentageError, ) -from torchmetrics.retrieval import ( # noqa: E402 +from torchmetrics.retrieval import ( RetrievalFallOut, RetrievalHitRate, RetrievalMAP, @@ -85,7 +85,7 @@ RetrievalRecallAtFixedPrecision, RetrievalRPrecision, ) -from torchmetrics.text import ( # noqa: E402 +from torchmetrics.text import ( BLEUScore, CharErrorRate, CHRFScore, @@ -99,7 +99,7 @@ WordInfoLost, WordInfoPreserved, ) -from torchmetrics.wrappers import ( # noqa: E402 +from torchmetrics.wrappers import ( BootStrapper, ClasswiseWrapper, MetricTracker, diff --git a/src/torchmetrics/audio/__init__.py b/src/torchmetrics/audio/__init__.py index 152228ac36a..22bf1ca1496 100644 --- a/src/torchmetrics/audio/__init__.py +++ b/src/torchmetrics/audio/__init__.py @@ -11,13 +11,13 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.audio.pit import PermutationInvariantTraining # noqa: F401 -from torchmetrics.audio.sdr import ScaleInvariantSignalDistortionRatio, SignalDistortionRatio # noqa: F401 -from torchmetrics.audio.snr import ScaleInvariantSignalNoiseRatio, SignalNoiseRatio # noqa: F401 +from torchmetrics.audio.pit import PermutationInvariantTraining +from torchmetrics.audio.sdr import ScaleInvariantSignalDistortionRatio, SignalDistortionRatio +from torchmetrics.audio.snr import ScaleInvariantSignalNoiseRatio, SignalNoiseRatio from torchmetrics.utilities.imports import _PESQ_AVAILABLE, _PYSTOI_AVAILABLE if _PESQ_AVAILABLE: - from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality # noqa: F401 + from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality if _PYSTOI_AVAILABLE: - from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility # noqa: F401 + from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility diff --git a/src/torchmetrics/detection/__init__.py b/src/torchmetrics/detection/__init__.py index e8a6d678856..00f86bd8ea8 100644 --- a/src/torchmetrics/detection/__init__.py +++ b/src/torchmetrics/detection/__init__.py @@ -14,4 +14,4 @@ from torchmetrics.utilities.imports import _TORCHVISION_GREATER_EQUAL_0_8 if _TORCHVISION_GREATER_EQUAL_0_8: - from torchmetrics.detection.mean_ap import MeanAveragePrecision # noqa: F401 + from torchmetrics.detection.mean_ap import MeanAveragePrecision diff --git a/src/torchmetrics/detection/mean_ap.py b/src/torchmetrics/detection/mean_ap.py index 0a0daf4aa89..c61a0fa4d30 100644 --- a/src/torchmetrics/detection/mean_ap.py +++ b/src/torchmetrics/detection/mean_ap.py @@ -861,7 +861,7 @@ def __calculate_recall_precision_scores( det_matches = torch.cat([e["dtMatches"][:, :max_det] for e in img_eval_cls_bbox], axis=1)[:, inds] det_ignore = torch.cat([e["dtIgnore"][:, :max_det] for e in img_eval_cls_bbox], axis=1)[:, inds] gt_ignore = torch.cat([e["gtIgnore"] for e in img_eval_cls_bbox]) - npig = torch.count_nonzero(gt_ignore == False) # noqa: E712 + npig = torch.count_nonzero(gt_ignore == False) if npig == 0: return recall, precision, scores tps = torch.logical_and(det_matches, torch.logical_not(det_ignore)) diff --git a/src/torchmetrics/functional/__init__.py b/src/torchmetrics/functional/__init__.py index c8fce9ee422..bb4bf15a9f5 100644 --- a/src/torchmetrics/functional/__init__.py +++ b/src/torchmetrics/functional/__init__.py @@ -95,8 +95,8 @@ from torchmetrics.utilities.imports import _TRANSFORMERS_AVAILABLE if _TRANSFORMERS_AVAILABLE: - from torchmetrics.functional.text.bert import bert_score # noqa: F401 - from torchmetrics.functional.text.infolm import infolm # noqa: F401 + from torchmetrics.functional.text.bert import bert_score + from torchmetrics.functional.text.infolm import infolm __all__ = [ "accuracy", diff --git a/src/torchmetrics/functional/audio/__init__.py b/src/torchmetrics/functional/audio/__init__.py index 67dc239989f..415f1e49f42 100644 --- a/src/torchmetrics/functional/audio/__init__.py +++ b/src/torchmetrics/functional/audio/__init__.py @@ -11,16 +11,16 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.audio.pit import permutation_invariant_training, pit_permutate # noqa: F401 -from torchmetrics.functional.audio.sdr import ( # noqa: F401 +from torchmetrics.functional.audio.pit import permutation_invariant_training, pit_permutate +from torchmetrics.functional.audio.sdr import ( scale_invariant_signal_distortion_ratio, signal_distortion_ratio, ) -from torchmetrics.functional.audio.snr import scale_invariant_signal_noise_ratio, signal_noise_ratio # noqa: F401 +from torchmetrics.functional.audio.snr import scale_invariant_signal_noise_ratio, signal_noise_ratio from torchmetrics.utilities.imports import _PESQ_AVAILABLE, _PYSTOI_AVAILABLE if _PESQ_AVAILABLE: - from torchmetrics.functional.audio.pesq import perceptual_evaluation_speech_quality # noqa: F401 + from torchmetrics.functional.audio.pesq import perceptual_evaluation_speech_quality if _PYSTOI_AVAILABLE: - from torchmetrics.functional.audio.stoi import short_time_objective_intelligibility # noqa: F401 + from torchmetrics.functional.audio.stoi import short_time_objective_intelligibility diff --git a/src/torchmetrics/functional/classification/__init__.py b/src/torchmetrics/functional/classification/__init__.py index 6aa93dd9261..e2acad5877c 100644 --- a/src/torchmetrics/functional/classification/__init__.py +++ b/src/torchmetrics/functional/classification/__init__.py @@ -11,47 +11,47 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.classification.accuracy import ( # noqa: F401 +from torchmetrics.functional.classification.accuracy import ( accuracy, binary_accuracy, multiclass_accuracy, multilabel_accuracy, ) -from torchmetrics.functional.classification.auroc import ( # noqa: F401 +from torchmetrics.functional.classification.auroc import ( auroc, binary_auroc, multiclass_auroc, multilabel_auroc, ) -from torchmetrics.functional.classification.average_precision import ( # noqa: F401 +from torchmetrics.functional.classification.average_precision import ( average_precision, binary_average_precision, multiclass_average_precision, multilabel_average_precision, ) -from torchmetrics.functional.classification.calibration_error import ( # noqa: F401 +from torchmetrics.functional.classification.calibration_error import ( binary_calibration_error, calibration_error, multiclass_calibration_error, ) -from torchmetrics.functional.classification.cohen_kappa import ( # noqa: F401 +from torchmetrics.functional.classification.cohen_kappa import ( binary_cohen_kappa, cohen_kappa, multiclass_cohen_kappa, ) -from torchmetrics.functional.classification.confusion_matrix import ( # noqa: F401 +from torchmetrics.functional.classification.confusion_matrix import ( binary_confusion_matrix, confusion_matrix, multiclass_confusion_matrix, multilabel_confusion_matrix, ) -from torchmetrics.functional.classification.dice import dice # noqa: F401 -from torchmetrics.functional.classification.exact_match import ( # noqa: F401 +from torchmetrics.functional.classification.dice import dice +from torchmetrics.functional.classification.exact_match import ( exact_match, multiclass_exact_match, multilabel_exact_match, ) -from torchmetrics.functional.classification.f_beta import ( # noqa: F401 +from torchmetrics.functional.classification.f_beta import ( binary_f1_score, binary_fbeta_score, f1_score, @@ -61,30 +61,30 @@ multilabel_f1_score, multilabel_fbeta_score, ) -from torchmetrics.functional.classification.hamming import ( # noqa: F401 +from torchmetrics.functional.classification.hamming import ( binary_hamming_distance, hamming_distance, multiclass_hamming_distance, multilabel_hamming_distance, ) -from torchmetrics.functional.classification.hinge import ( # noqa: F401 +from torchmetrics.functional.classification.hinge import ( binary_hinge_loss, hinge_loss, multiclass_hinge_loss, ) -from torchmetrics.functional.classification.jaccard import ( # noqa: F401 +from torchmetrics.functional.classification.jaccard import ( binary_jaccard_index, jaccard_index, multiclass_jaccard_index, multilabel_jaccard_index, ) -from torchmetrics.functional.classification.matthews_corrcoef import ( # noqa: F401 +from torchmetrics.functional.classification.matthews_corrcoef import ( binary_matthews_corrcoef, matthews_corrcoef, multiclass_matthews_corrcoef, multilabel_matthews_corrcoef, ) -from torchmetrics.functional.classification.precision_recall import ( # noqa: F401 +from torchmetrics.functional.classification.precision_recall import ( binary_precision, binary_recall, multiclass_precision, @@ -94,30 +94,30 @@ precision, recall, ) -from torchmetrics.functional.classification.precision_recall_curve import ( # noqa: F401 +from torchmetrics.functional.classification.precision_recall_curve import ( binary_precision_recall_curve, multiclass_precision_recall_curve, multilabel_precision_recall_curve, precision_recall_curve, ) -from torchmetrics.functional.classification.ranking import ( # noqa: F401 +from torchmetrics.functional.classification.ranking import ( multilabel_coverage_error, multilabel_ranking_average_precision, multilabel_ranking_loss, ) -from torchmetrics.functional.classification.recall_at_fixed_precision import ( # noqa: F401 +from torchmetrics.functional.classification.recall_at_fixed_precision import ( binary_recall_at_fixed_precision, multiclass_recall_at_fixed_precision, multilabel_recall_at_fixed_precision, ) -from torchmetrics.functional.classification.roc import binary_roc, multiclass_roc, multilabel_roc, roc # noqa: F401 -from torchmetrics.functional.classification.specificity import ( # noqa: F401 +from torchmetrics.functional.classification.roc import binary_roc, multiclass_roc, multilabel_roc, roc +from torchmetrics.functional.classification.specificity import ( binary_specificity, multiclass_specificity, multilabel_specificity, specificity, ) -from torchmetrics.functional.classification.stat_scores import ( # noqa: F401 +from torchmetrics.functional.classification.stat_scores import ( binary_stat_scores, multiclass_stat_scores, multilabel_stat_scores, diff --git a/src/torchmetrics/functional/image/__init__.py b/src/torchmetrics/functional/image/__init__.py index c3c53501018..e4fae6c4dd9 100644 --- a/src/torchmetrics/functional/image/__init__.py +++ b/src/torchmetrics/functional/image/__init__.py @@ -11,14 +11,14 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.image.d_lambda import spectral_distortion_index # noqa: F401 -from torchmetrics.functional.image.ergas import error_relative_global_dimensionless_synthesis # noqa: F401 -from torchmetrics.functional.image.gradients import image_gradients # noqa: F401 -from torchmetrics.functional.image.psnr import peak_signal_noise_ratio # noqa: F401 -from torchmetrics.functional.image.sam import spectral_angle_mapper # noqa: F401 -from torchmetrics.functional.image.ssim import ( # noqa: F401 +from torchmetrics.functional.image.d_lambda import spectral_distortion_index +from torchmetrics.functional.image.ergas import error_relative_global_dimensionless_synthesis +from torchmetrics.functional.image.gradients import image_gradients +from torchmetrics.functional.image.psnr import peak_signal_noise_ratio +from torchmetrics.functional.image.sam import spectral_angle_mapper +from torchmetrics.functional.image.ssim import ( multiscale_structural_similarity_index_measure, structural_similarity_index_measure, ) -from torchmetrics.functional.image.tv import total_variation # noqa: F401 -from torchmetrics.functional.image.uqi import universal_image_quality_index # noqa: F401 +from torchmetrics.functional.image.tv import total_variation +from torchmetrics.functional.image.uqi import universal_image_quality_index diff --git a/src/torchmetrics/functional/multimodal/__init__.py b/src/torchmetrics/functional/multimodal/__init__.py index 42b82326afe..66c9dbbf020 100644 --- a/src/torchmetrics/functional/multimodal/__init__.py +++ b/src/torchmetrics/functional/multimodal/__init__.py @@ -14,4 +14,4 @@ from torchmetrics.utilities.imports import _TRANSFORMERS_AVAILABLE if _TRANSFORMERS_AVAILABLE: - from torchmetrics.functional.multimodal.clip_score import clip_score # noqa: F401 + from torchmetrics.functional.multimodal.clip_score import clip_score diff --git a/src/torchmetrics/functional/nominal/__init__.py b/src/torchmetrics/functional/nominal/__init__.py index c12c93ac700..0ae6cdf2b23 100644 --- a/src/torchmetrics/functional/nominal/__init__.py +++ b/src/torchmetrics/functional/nominal/__init__.py @@ -11,10 +11,10 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.nominal.cramers import cramers_v, cramers_v_matrix # noqa: F401 -from torchmetrics.functional.nominal.pearson import ( # noqa: F401 +from torchmetrics.functional.nominal.cramers import cramers_v, cramers_v_matrix +from torchmetrics.functional.nominal.pearson import ( pearsons_contingency_coefficient, pearsons_contingency_coefficient_matrix, ) -from torchmetrics.functional.nominal.theils_u import theils_u, theils_u_matrix # noqa: F401 -from torchmetrics.functional.nominal.tschuprows import tschuprows_t, tschuprows_t_matrix # noqa: F401 +from torchmetrics.functional.nominal.theils_u import theils_u, theils_u_matrix +from torchmetrics.functional.nominal.tschuprows import tschuprows_t, tschuprows_t_matrix diff --git a/src/torchmetrics/functional/pairwise/__init__.py b/src/torchmetrics/functional/pairwise/__init__.py index 0027a4aad7d..7bcca04d25b 100644 --- a/src/torchmetrics/functional/pairwise/__init__.py +++ b/src/torchmetrics/functional/pairwise/__init__.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.pairwise.cosine import pairwise_cosine_similarity # noqa: F401 -from torchmetrics.functional.pairwise.euclidean import pairwise_euclidean_distance # noqa: F401 -from torchmetrics.functional.pairwise.linear import pairwise_linear_similarity # noqa: F401 -from torchmetrics.functional.pairwise.manhattan import pairwise_manhattan_distance # noqa: F401 +from torchmetrics.functional.pairwise.cosine import pairwise_cosine_similarity +from torchmetrics.functional.pairwise.euclidean import pairwise_euclidean_distance +from torchmetrics.functional.pairwise.linear import pairwise_linear_similarity +from torchmetrics.functional.pairwise.manhattan import pairwise_manhattan_distance diff --git a/src/torchmetrics/functional/regression/__init__.py b/src/torchmetrics/functional/regression/__init__.py index 4cef75a985b..84d73c6b555 100644 --- a/src/torchmetrics/functional/regression/__init__.py +++ b/src/torchmetrics/functional/regression/__init__.py @@ -11,18 +11,18 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.regression.concordance import concordance_corrcoef # noqa: F401 -from torchmetrics.functional.regression.cosine_similarity import cosine_similarity # noqa: F401 -from torchmetrics.functional.regression.explained_variance import explained_variance # noqa: F401 -from torchmetrics.functional.regression.kendall import kendall_rank_corrcoef # noqa: F401 -from torchmetrics.functional.regression.kl_divergence import kl_divergence # noqa: F401 -from torchmetrics.functional.regression.log_cosh import log_cosh_error # noqa: F401 -from torchmetrics.functional.regression.log_mse import mean_squared_log_error # noqa: F401 -from torchmetrics.functional.regression.mae import mean_absolute_error # noqa: F401 -from torchmetrics.functional.regression.mape import mean_absolute_percentage_error # noqa: F401 -from torchmetrics.functional.regression.mse import mean_squared_error # noqa: F401 -from torchmetrics.functional.regression.pearson import pearson_corrcoef # noqa: F401 -from torchmetrics.functional.regression.r2 import r2_score # noqa: F401 -from torchmetrics.functional.regression.spearman import spearman_corrcoef # noqa: F401 -from torchmetrics.functional.regression.tweedie_deviance import tweedie_deviance_score # noqa: F401 -from torchmetrics.functional.regression.wmape import weighted_mean_absolute_percentage_error # noqa: F401 +from torchmetrics.functional.regression.concordance import concordance_corrcoef +from torchmetrics.functional.regression.cosine_similarity import cosine_similarity +from torchmetrics.functional.regression.explained_variance import explained_variance +from torchmetrics.functional.regression.kendall import kendall_rank_corrcoef +from torchmetrics.functional.regression.kl_divergence import kl_divergence +from torchmetrics.functional.regression.log_cosh import log_cosh_error +from torchmetrics.functional.regression.log_mse import mean_squared_log_error +from torchmetrics.functional.regression.mae import mean_absolute_error +from torchmetrics.functional.regression.mape import mean_absolute_percentage_error +from torchmetrics.functional.regression.mse import mean_squared_error +from torchmetrics.functional.regression.pearson import pearson_corrcoef +from torchmetrics.functional.regression.r2 import r2_score +from torchmetrics.functional.regression.spearman import spearman_corrcoef +from torchmetrics.functional.regression.tweedie_deviance import tweedie_deviance_score +from torchmetrics.functional.regression.wmape import weighted_mean_absolute_percentage_error diff --git a/src/torchmetrics/functional/retrieval/__init__.py b/src/torchmetrics/functional/retrieval/__init__.py index b50a1f06ce5..4e5d448cace 100644 --- a/src/torchmetrics/functional/retrieval/__init__.py +++ b/src/torchmetrics/functional/retrieval/__init__.py @@ -12,12 +12,12 @@ # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.retrieval.average_precision import retrieval_average_precision # noqa: F401 -from torchmetrics.functional.retrieval.fall_out import retrieval_fall_out # noqa: F401 -from torchmetrics.functional.retrieval.hit_rate import retrieval_hit_rate # noqa: F401 -from torchmetrics.functional.retrieval.ndcg import retrieval_normalized_dcg # noqa: F401 -from torchmetrics.functional.retrieval.precision import retrieval_precision # noqa: F401 -from torchmetrics.functional.retrieval.precision_recall_curve import retrieval_precision_recall_curve # noqa: F401 -from torchmetrics.functional.retrieval.r_precision import retrieval_r_precision # noqa: F401 -from torchmetrics.functional.retrieval.recall import retrieval_recall # noqa: F401 -from torchmetrics.functional.retrieval.reciprocal_rank import retrieval_reciprocal_rank # noqa: F401 +from torchmetrics.functional.retrieval.average_precision import retrieval_average_precision +from torchmetrics.functional.retrieval.fall_out import retrieval_fall_out +from torchmetrics.functional.retrieval.hit_rate import retrieval_hit_rate +from torchmetrics.functional.retrieval.ndcg import retrieval_normalized_dcg +from torchmetrics.functional.retrieval.precision import retrieval_precision +from torchmetrics.functional.retrieval.precision_recall_curve import retrieval_precision_recall_curve +from torchmetrics.functional.retrieval.r_precision import retrieval_r_precision +from torchmetrics.functional.retrieval.recall import retrieval_recall +from torchmetrics.functional.retrieval.reciprocal_rank import retrieval_reciprocal_rank diff --git a/src/torchmetrics/functional/text/__init__.py b/src/torchmetrics/functional/text/__init__.py index afb371ba79e..5e8019738a4 100644 --- a/src/torchmetrics/functional/text/__init__.py +++ b/src/torchmetrics/functional/text/__init__.py @@ -12,23 +12,23 @@ # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.functional.text.bleu import bleu_score # noqa: F401 -from torchmetrics.functional.text.cer import char_error_rate # noqa: F401 -from torchmetrics.functional.text.chrf import chrf_score # noqa: F401 -from torchmetrics.functional.text.eed import extended_edit_distance # noqa: F401 -from torchmetrics.functional.text.mer import match_error_rate # noqa: F401 -from torchmetrics.functional.text.perplexity import perplexity # noqa: F401 -from torchmetrics.functional.text.sacre_bleu import sacre_bleu_score # noqa: F401 -from torchmetrics.functional.text.squad import squad # noqa: F401 -from torchmetrics.functional.text.ter import translation_edit_rate # noqa: F401 -from torchmetrics.functional.text.wer import word_error_rate # noqa: F401 -from torchmetrics.functional.text.wil import word_information_lost # noqa: F401 -from torchmetrics.functional.text.wip import word_information_preserved # noqa: F401 +from torchmetrics.functional.text.bleu import bleu_score +from torchmetrics.functional.text.cer import char_error_rate +from torchmetrics.functional.text.chrf import chrf_score +from torchmetrics.functional.text.eed import extended_edit_distance +from torchmetrics.functional.text.mer import match_error_rate +from torchmetrics.functional.text.perplexity import perplexity +from torchmetrics.functional.text.sacre_bleu import sacre_bleu_score +from torchmetrics.functional.text.squad import squad +from torchmetrics.functional.text.ter import translation_edit_rate +from torchmetrics.functional.text.wer import word_error_rate +from torchmetrics.functional.text.wil import word_information_lost +from torchmetrics.functional.text.wip import word_information_preserved from torchmetrics.utilities.imports import _NLTK_AVAILABLE, _TRANSFORMERS_AVAILABLE if _TRANSFORMERS_AVAILABLE: - from torchmetrics.functional.text.bert import bert_score # noqa: F401 - from torchmetrics.functional.text.infolm import infolm # noqa: F401 + from torchmetrics.functional.text.bert import bert_score + from torchmetrics.functional.text.infolm import infolm if _NLTK_AVAILABLE: - from torchmetrics.functional.text.rouge import rouge_score # noqa: F401 + from torchmetrics.functional.text.rouge import rouge_score diff --git a/src/torchmetrics/image/__init__.py b/src/torchmetrics/image/__init__.py index 781c28c8ab4..e12bfa66b76 100644 --- a/src/torchmetrics/image/__init__.py +++ b/src/torchmetrics/image/__init__.py @@ -11,23 +11,23 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.image.d_lambda import SpectralDistortionIndex # noqa: F401 -from torchmetrics.image.ergas import ErrorRelativeGlobalDimensionlessSynthesis # noqa: F401 -from torchmetrics.image.psnr import PeakSignalNoiseRatio # noqa: F401 -from torchmetrics.image.sam import SpectralAngleMapper # noqa: F401 -from torchmetrics.image.ssim import ( # noqa: F401 +from torchmetrics.image.d_lambda import SpectralDistortionIndex +from torchmetrics.image.ergas import ErrorRelativeGlobalDimensionlessSynthesis +from torchmetrics.image.psnr import PeakSignalNoiseRatio +from torchmetrics.image.sam import SpectralAngleMapper +from torchmetrics.image.ssim import ( MultiScaleStructuralSimilarityIndexMeasure, StructuralSimilarityIndexMeasure, ) -from torchmetrics.image.uqi import UniversalImageQualityIndex # noqa: F401 +from torchmetrics.image.uqi import UniversalImageQualityIndex from torchmetrics.utilities.imports import _LPIPS_AVAILABLE, _TORCH_FIDELITY_AVAILABLE if _TORCH_FIDELITY_AVAILABLE: - from torchmetrics.image.fid import FrechetInceptionDistance # noqa: F401 - from torchmetrics.image.inception import InceptionScore # noqa: F401 - from torchmetrics.image.kid import KernelInceptionDistance # noqa: F401 + from torchmetrics.image.fid import FrechetInceptionDistance + from torchmetrics.image.inception import InceptionScore + from torchmetrics.image.kid import KernelInceptionDistance if _LPIPS_AVAILABLE: - from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity # noqa: F401 + from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity -from torchmetrics.image.tv import TotalVariation # noqa: F401 +from torchmetrics.image.tv import TotalVariation diff --git a/src/torchmetrics/multimodal/__init__.py b/src/torchmetrics/multimodal/__init__.py index e5b4ad56ce2..fecf96965ae 100644 --- a/src/torchmetrics/multimodal/__init__.py +++ b/src/torchmetrics/multimodal/__init__.py @@ -14,4 +14,4 @@ from torchmetrics.utilities.imports import _TRANSFORMERS_AVAILABLE if _TRANSFORMERS_AVAILABLE: - from torchmetrics.multimodal.clip_score import CLIPScore # noqa: F401 + from torchmetrics.multimodal.clip_score import CLIPScore diff --git a/src/torchmetrics/nominal/__init__.py b/src/torchmetrics/nominal/__init__.py index d26572f4a27..88008ef8911 100644 --- a/src/torchmetrics/nominal/__init__.py +++ b/src/torchmetrics/nominal/__init__.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.nominal.cramers import CramersV # noqa: F401 -from torchmetrics.nominal.pearson import PearsonsContingencyCoefficient # noqa: F401 -from torchmetrics.nominal.theils_u import TheilsU # noqa: F401 -from torchmetrics.nominal.tschuprows import TschuprowsT # noqa: F401 +from torchmetrics.nominal.cramers import CramersV +from torchmetrics.nominal.pearson import PearsonsContingencyCoefficient +from torchmetrics.nominal.theils_u import TheilsU +from torchmetrics.nominal.tschuprows import TschuprowsT diff --git a/src/torchmetrics/regression/__init__.py b/src/torchmetrics/regression/__init__.py index 860a670eb3d..50ea063dd51 100644 --- a/src/torchmetrics/regression/__init__.py +++ b/src/torchmetrics/regression/__init__.py @@ -11,19 +11,19 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.regression.concordance import ConcordanceCorrCoef # noqa: F401 -from torchmetrics.regression.cosine_similarity import CosineSimilarity # noqa: F401 -from torchmetrics.regression.explained_variance import ExplainedVariance # noqa: F401 -from torchmetrics.regression.kendall import KendallRankCorrCoef # noqa: F401 -from torchmetrics.regression.kl_divergence import KLDivergence # noqa: F401 -from torchmetrics.regression.log_cosh import LogCoshError # noqa: F401 -from torchmetrics.regression.log_mse import MeanSquaredLogError # noqa: F401 -from torchmetrics.regression.mae import MeanAbsoluteError # noqa: F401 -from torchmetrics.regression.mape import MeanAbsolutePercentageError # noqa: F401 -from torchmetrics.regression.mse import MeanSquaredError # noqa: F401 -from torchmetrics.regression.pearson import PearsonCorrCoef # noqa: F401 -from torchmetrics.regression.r2 import R2Score # noqa: F401 -from torchmetrics.regression.spearman import SpearmanCorrCoef # noqa: F401 -from torchmetrics.regression.symmetric_mape import SymmetricMeanAbsolutePercentageError # noqa: F401 -from torchmetrics.regression.tweedie_deviance import TweedieDevianceScore # noqa: F401 -from torchmetrics.regression.wmape import WeightedMeanAbsolutePercentageError # noqa: F401 +from torchmetrics.regression.concordance import ConcordanceCorrCoef +from torchmetrics.regression.cosine_similarity import CosineSimilarity +from torchmetrics.regression.explained_variance import ExplainedVariance +from torchmetrics.regression.kendall import KendallRankCorrCoef +from torchmetrics.regression.kl_divergence import KLDivergence +from torchmetrics.regression.log_cosh import LogCoshError +from torchmetrics.regression.log_mse import MeanSquaredLogError +from torchmetrics.regression.mae import MeanAbsoluteError +from torchmetrics.regression.mape import MeanAbsolutePercentageError +from torchmetrics.regression.mse import MeanSquaredError +from torchmetrics.regression.pearson import PearsonCorrCoef +from torchmetrics.regression.r2 import R2Score +from torchmetrics.regression.spearman import SpearmanCorrCoef +from torchmetrics.regression.symmetric_mape import SymmetricMeanAbsolutePercentageError +from torchmetrics.regression.tweedie_deviance import TweedieDevianceScore +from torchmetrics.regression.wmape import WeightedMeanAbsolutePercentageError diff --git a/src/torchmetrics/retrieval/__init__.py b/src/torchmetrics/retrieval/__init__.py index dff16a7d19f..63134335ba3 100644 --- a/src/torchmetrics/retrieval/__init__.py +++ b/src/torchmetrics/retrieval/__init__.py @@ -11,16 +11,16 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.retrieval.average_precision import RetrievalMAP # noqa: F401 -from torchmetrics.retrieval.base import RetrievalMetric # noqa: F401 -from torchmetrics.retrieval.fall_out import RetrievalFallOut # noqa: F401 -from torchmetrics.retrieval.hit_rate import RetrievalHitRate # noqa: F401 -from torchmetrics.retrieval.ndcg import RetrievalNormalizedDCG # noqa: F401 -from torchmetrics.retrieval.precision import RetrievalPrecision # noqa: F401 -from torchmetrics.retrieval.precision_recall_curve import ( # noqa: F401 +from torchmetrics.retrieval.average_precision import RetrievalMAP +from torchmetrics.retrieval.base import RetrievalMetric +from torchmetrics.retrieval.fall_out import RetrievalFallOut +from torchmetrics.retrieval.hit_rate import RetrievalHitRate +from torchmetrics.retrieval.ndcg import RetrievalNormalizedDCG +from torchmetrics.retrieval.precision import RetrievalPrecision +from torchmetrics.retrieval.precision_recall_curve import ( RetrievalPrecisionRecallCurve, RetrievalRecallAtFixedPrecision, ) -from torchmetrics.retrieval.r_precision import RetrievalRPrecision # noqa: F401 -from torchmetrics.retrieval.recall import RetrievalRecall # noqa: F401 -from torchmetrics.retrieval.reciprocal_rank import RetrievalMRR # noqa: F401 +from torchmetrics.retrieval.r_precision import RetrievalRPrecision +from torchmetrics.retrieval.recall import RetrievalRecall +from torchmetrics.retrieval.reciprocal_rank import RetrievalMRR diff --git a/src/torchmetrics/text/__init__.py b/src/torchmetrics/text/__init__.py index 072ca55d3fc..180138c6189 100644 --- a/src/torchmetrics/text/__init__.py +++ b/src/torchmetrics/text/__init__.py @@ -11,23 +11,23 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.text.bleu import BLEUScore # noqa: F401 -from torchmetrics.text.cer import CharErrorRate # noqa: F401 -from torchmetrics.text.chrf import CHRFScore # noqa: F401 -from torchmetrics.text.eed import ExtendedEditDistance # noqa: F401 -from torchmetrics.text.mer import MatchErrorRate # noqa: F401 -from torchmetrics.text.perplexity import Perplexity # noqa: F401 -from torchmetrics.text.sacre_bleu import SacreBLEUScore # noqa: F401 -from torchmetrics.text.squad import SQuAD # noqa: F401 -from torchmetrics.text.ter import TranslationEditRate # noqa: F401 -from torchmetrics.text.wer import WordErrorRate # noqa: F401 -from torchmetrics.text.wil import WordInfoLost # noqa: F401 -from torchmetrics.text.wip import WordInfoPreserved # noqa: F401 +from torchmetrics.text.bleu import BLEUScore +from torchmetrics.text.cer import CharErrorRate +from torchmetrics.text.chrf import CHRFScore +from torchmetrics.text.eed import ExtendedEditDistance +from torchmetrics.text.mer import MatchErrorRate +from torchmetrics.text.perplexity import Perplexity +from torchmetrics.text.sacre_bleu import SacreBLEUScore +from torchmetrics.text.squad import SQuAD +from torchmetrics.text.ter import TranslationEditRate +from torchmetrics.text.wer import WordErrorRate +from torchmetrics.text.wil import WordInfoLost +from torchmetrics.text.wip import WordInfoPreserved from torchmetrics.utilities.imports import _NLTK_AVAILABLE, _TRANSFORMERS_AVAILABLE if _TRANSFORMERS_AVAILABLE: - from torchmetrics.text.bert import BERTScore # noqa: F401 - from torchmetrics.text.infolm import InfoLM # noqa: F401 + from torchmetrics.text.bert import BERTScore + from torchmetrics.text.infolm import InfoLM if _NLTK_AVAILABLE: - from torchmetrics.text.rouge import ROUGEScore # noqa: F401 + from torchmetrics.text.rouge import ROUGEScore diff --git a/src/torchmetrics/utilities/__init__.py b/src/torchmetrics/utilities/__init__.py index 231355a56db..e3fc6acd738 100644 --- a/src/torchmetrics/utilities/__init__.py +++ b/src/torchmetrics/utilities/__init__.py @@ -1,4 +1,4 @@ -from torchmetrics.utilities.checks import check_forward_full_state_property # noqa: F401 -from torchmetrics.utilities.data import apply_to_collection # noqa: F401 -from torchmetrics.utilities.distributed import class_reduce, reduce # noqa: F401 -from torchmetrics.utilities.prints import _future_warning, rank_zero_debug, rank_zero_info, rank_zero_warn # noqa: F401 +from torchmetrics.utilities.checks import check_forward_full_state_property +from torchmetrics.utilities.data import apply_to_collection +from torchmetrics.utilities.distributed import class_reduce, reduce +from torchmetrics.utilities.prints import _future_warning, rank_zero_debug, rank_zero_info, rank_zero_warn diff --git a/src/torchmetrics/utilities/imports.py b/src/torchmetrics/utilities/imports.py index b05c1bbb04b..fe1fb58d30b 100644 --- a/src/torchmetrics/utilities/imports.py +++ b/src/torchmetrics/utilities/imports.py @@ -13,7 +13,7 @@ # limitations under the License. """Import utilities.""" import operator -from collections import OrderedDict # noqa: F401 +from collections import OrderedDict from functools import lru_cache from importlib import import_module from importlib.util import find_spec diff --git a/src/torchmetrics/wrappers/__init__.py b/src/torchmetrics/wrappers/__init__.py index 98dbb137b19..379ad9fc869 100644 --- a/src/torchmetrics/wrappers/__init__.py +++ b/src/torchmetrics/wrappers/__init__.py @@ -11,8 +11,8 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from torchmetrics.wrappers.bootstrapping import BootStrapper # noqa: F401 -from torchmetrics.wrappers.classwise import ClasswiseWrapper # noqa: F401 -from torchmetrics.wrappers.minmax import MinMaxMetric # noqa: F401 -from torchmetrics.wrappers.multioutput import MultioutputWrapper # noqa: F401 -from torchmetrics.wrappers.tracker import MetricTracker # noqa: F401 +from torchmetrics.wrappers.bootstrapping import BootStrapper +from torchmetrics.wrappers.classwise import ClasswiseWrapper +from torchmetrics.wrappers.minmax import MinMaxMetric +from torchmetrics.wrappers.multioutput import MultioutputWrapper +from torchmetrics.wrappers.tracker import MetricTracker diff --git a/tests/unittests/__init__.py b/tests/unittests/__init__.py index c0f0fd23aef..bdf33a91d2f 100644 --- a/tests/unittests/__init__.py +++ b/tests/unittests/__init__.py @@ -1,6 +1,6 @@ import os.path -from unittests.helpers.testers import BATCH_SIZE, NUM_BATCHES, NUM_PROCESSES, DummyMetric, MetricTester # noqa: F401 +from unittests.helpers.testers import BATCH_SIZE, NUM_BATCHES, NUM_PROCESSES, DummyMetric, MetricTester _PATH_TESTS = os.path.dirname(__file__) _PATH_ROOT = os.path.dirname(_PATH_TESTS)