From 0e87d9adb59f23e76b50fe9cb5eb44edc08cad86 Mon Sep 17 00:00:00 2001 From: Jirka Date: Thu, 13 Jan 2022 20:21:09 +0100 Subject: [PATCH] move to prints --- torchmetrics/audio/pit.py | 2 +- torchmetrics/audio/sdr.py | 2 +- torchmetrics/audio/si_sdr.py | 2 +- torchmetrics/audio/si_snr.py | 2 +- torchmetrics/audio/snr.py | 2 +- torchmetrics/classification/f_beta.py | 2 +- torchmetrics/classification/hinge.py | 2 +- torchmetrics/classification/iou.py | 2 +- .../classification/matthews_corrcoef.py | 2 +- torchmetrics/functional/audio/pit.py | 2 +- torchmetrics/functional/audio/sdr.py | 3 +-- torchmetrics/functional/audio/si_sdr.py | 2 +- torchmetrics/functional/audio/si_snr.py | 2 +- torchmetrics/functional/audio/snr.py | 2 +- .../functional/classification/f_beta.py | 2 +- .../functional/classification/hinge.py | 2 +- torchmetrics/functional/classification/iou.py | 2 +- torchmetrics/functional/image/ms_ssim.py | 2 +- torchmetrics/functional/image/psnr.py | 3 +-- torchmetrics/functional/image/ssim.py | 2 +- torchmetrics/functional/pairwise/manhatten.py | 2 +- torchmetrics/functional/text/bert.py | 2 +- torchmetrics/functional/text/bleu.py | 2 +- torchmetrics/functional/text/cer.py | 2 +- torchmetrics/functional/text/sacre_bleu.py | 2 +- torchmetrics/functional/text/wer.py | 2 +- torchmetrics/image/psnr.py | 3 +-- torchmetrics/image/ssim.py | 3 +-- torchmetrics/regression/pearson.py | 2 +- torchmetrics/regression/spearman.py | 3 +-- torchmetrics/text/bert.py | 2 +- torchmetrics/text/bleu.py | 2 +- torchmetrics/text/cer.py | 2 +- torchmetrics/text/sacre_bleu.py | 2 +- torchmetrics/text/wer.py | 2 +- torchmetrics/utilities/__init__.py | 3 +-- torchmetrics/utilities/prints.py | 5 ++++- torchmetrics/utilities/warn.py | 18 ------------------ 38 files changed, 40 insertions(+), 61 deletions(-) delete mode 100644 torchmetrics/utilities/warn.py diff --git a/torchmetrics/audio/pit.py b/torchmetrics/audio/pit.py index 7512ca97f7d..24acf6c1bae 100644 --- a/torchmetrics/audio/pit.py +++ b/torchmetrics/audio/pit.py @@ -18,7 +18,7 @@ from torchmetrics.functional.audio.pit import permutation_invariant_training from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class PermutationInvariantTraining(Metric): diff --git a/torchmetrics/audio/sdr.py b/torchmetrics/audio/sdr.py index 9a59bf6ef1b..2acb4a98da5 100644 --- a/torchmetrics/audio/sdr.py +++ b/torchmetrics/audio/sdr.py @@ -18,8 +18,8 @@ from torchmetrics.functional.audio.sdr import scale_invariant_signal_distortion_ratio, signal_distortion_ratio from torchmetrics.metric import Metric +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.imports import _FAST_BSS_EVAL_AVAILABLE -from torchmetrics.utilities.warn import _future_warning class SignalDistortionRatio(Metric): diff --git a/torchmetrics/audio/si_sdr.py b/torchmetrics/audio/si_sdr.py index fb44b20ce13..a77a466f586 100644 --- a/torchmetrics/audio/si_sdr.py +++ b/torchmetrics/audio/si_sdr.py @@ -17,7 +17,7 @@ from torch import Tensor from torchmetrics.audio.sdr import ScaleInvariantSignalDistortionRatio -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class SI_SDR(ScaleInvariantSignalDistortionRatio): diff --git a/torchmetrics/audio/si_snr.py b/torchmetrics/audio/si_snr.py index 7a0292eb4a2..7f6388d36d3 100644 --- a/torchmetrics/audio/si_snr.py +++ b/torchmetrics/audio/si_snr.py @@ -17,7 +17,7 @@ from torch import Tensor from torchmetrics.audio.snr import ScaleInvariantSignalNoiseRatio -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class SI_SNR(ScaleInvariantSignalNoiseRatio): diff --git a/torchmetrics/audio/snr.py b/torchmetrics/audio/snr.py index 0b5c5e767fb..0a2532e564d 100644 --- a/torchmetrics/audio/snr.py +++ b/torchmetrics/audio/snr.py @@ -18,7 +18,7 @@ from torchmetrics.functional.audio.snr import scale_invariant_signal_noise_ratio, snr from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class SignalNoiseRatio(Metric): diff --git a/torchmetrics/classification/f_beta.py b/torchmetrics/classification/f_beta.py index 63d37e3163a..0a716964453 100644 --- a/torchmetrics/classification/f_beta.py +++ b/torchmetrics/classification/f_beta.py @@ -19,8 +19,8 @@ from torchmetrics.classification.stat_scores import StatScores from torchmetrics.functional.classification.f_beta import _fbeta_compute +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.enums import AverageMethod -from torchmetrics.utilities.warn import _future_warning class FBetaScore(StatScores): diff --git a/torchmetrics/classification/hinge.py b/torchmetrics/classification/hinge.py index 5cb026909e5..04b4736cb22 100644 --- a/torchmetrics/classification/hinge.py +++ b/torchmetrics/classification/hinge.py @@ -18,7 +18,7 @@ from torchmetrics.functional.classification.hinge import MulticlassMode, _hinge_compute, _hinge_update from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class HingeLoss(Metric): diff --git a/torchmetrics/classification/iou.py b/torchmetrics/classification/iou.py index 11f61bf6631..d11a9afab83 100644 --- a/torchmetrics/classification/iou.py +++ b/torchmetrics/classification/iou.py @@ -17,7 +17,7 @@ from deprecate import deprecated, void from torchmetrics.classification.jaccard import JaccardIndex -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class IoU(JaccardIndex): diff --git a/torchmetrics/classification/matthews_corrcoef.py b/torchmetrics/classification/matthews_corrcoef.py index 174c8f88957..ffe96767666 100644 --- a/torchmetrics/classification/matthews_corrcoef.py +++ b/torchmetrics/classification/matthews_corrcoef.py @@ -22,7 +22,7 @@ _matthews_corrcoef_update, ) from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class MatthewsCorrCoef(Metric): diff --git a/torchmetrics/functional/audio/pit.py b/torchmetrics/functional/audio/pit.py index f2cf1ca30e1..3b0ebefbc01 100644 --- a/torchmetrics/functional/audio/pit.py +++ b/torchmetrics/functional/audio/pit.py @@ -19,9 +19,9 @@ from deprecate import deprecated, void from torch import Tensor +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.checks import _check_same_shape from torchmetrics.utilities.imports import _SCIPY_AVAILABLE -from torchmetrics.utilities.warn import _future_warning # _ps_dict: cache of permutations # it's necessary to cache it, otherwise it will consume a large amount of time diff --git a/torchmetrics/functional/audio/sdr.py b/torchmetrics/functional/audio/sdr.py index 7b5845f039d..85e92c100c9 100644 --- a/torchmetrics/functional/audio/sdr.py +++ b/torchmetrics/functional/audio/sdr.py @@ -18,7 +18,6 @@ from deprecate import deprecated, void from torchmetrics.utilities.imports import _FAST_BSS_EVAL_AVAILABLE, _TORCH_GREATER_EQUAL_1_8 -from torchmetrics.utilities.warn import _future_warning if _FAST_BSS_EVAL_AVAILABLE: if _TORCH_GREATER_EQUAL_1_8: @@ -44,7 +43,7 @@ from torch import Tensor -from torchmetrics.utilities import rank_zero_warn +from torchmetrics.utilities import _future_warning, rank_zero_warn from torchmetrics.utilities.checks import _check_same_shape diff --git a/torchmetrics/functional/audio/si_sdr.py b/torchmetrics/functional/audio/si_sdr.py index ca45f9d1e57..6ad04332f96 100644 --- a/torchmetrics/functional/audio/si_sdr.py +++ b/torchmetrics/functional/audio/si_sdr.py @@ -17,7 +17,7 @@ from torch import Tensor from torchmetrics.functional.audio import scale_invariant_signal_distortion_ratio -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning @deprecated( diff --git a/torchmetrics/functional/audio/si_snr.py b/torchmetrics/functional/audio/si_snr.py index 7a785f52893..2b63e4eb675 100644 --- a/torchmetrics/functional/audio/si_snr.py +++ b/torchmetrics/functional/audio/si_snr.py @@ -16,7 +16,7 @@ from torch import Tensor from torchmetrics.functional.audio.snr import scale_invariant_signal_noise_ratio -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning @deprecated(target=scale_invariant_signal_noise_ratio, deprecated_in="0.7", remove_in="0.8", stream=_future_warning) diff --git a/torchmetrics/functional/audio/snr.py b/torchmetrics/functional/audio/snr.py index ef7e07112fd..274ddbed700 100644 --- a/torchmetrics/functional/audio/snr.py +++ b/torchmetrics/functional/audio/snr.py @@ -17,8 +17,8 @@ from torch import Tensor from torchmetrics.functional.audio.sdr import scale_invariant_signal_distortion_ratio +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.checks import _check_same_shape -from torchmetrics.utilities.warn import _future_warning def signal_noise_ratio(preds: Tensor, target: Tensor, zero_mean: bool = False) -> Tensor: diff --git a/torchmetrics/functional/classification/f_beta.py b/torchmetrics/functional/classification/f_beta.py index a6d12c1da21..60571603b8a 100644 --- a/torchmetrics/functional/classification/f_beta.py +++ b/torchmetrics/functional/classification/f_beta.py @@ -18,9 +18,9 @@ from torch import Tensor from torchmetrics.functional.classification.stat_scores import _reduce_stat_scores, _stat_scores_update +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.enums import AverageMethod as AvgMethod from torchmetrics.utilities.enums import MDMCAverageMethod -from torchmetrics.utilities.warn import _future_warning def _safe_divide(num: Tensor, denom: Tensor) -> Tensor: diff --git a/torchmetrics/functional/classification/hinge.py b/torchmetrics/functional/classification/hinge.py index 190c1fa2d2e..649d3c8a5d6 100644 --- a/torchmetrics/functional/classification/hinge.py +++ b/torchmetrics/functional/classification/hinge.py @@ -17,10 +17,10 @@ from deprecate import deprecated, void from torch import Tensor, tensor +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.checks import _input_squeeze from torchmetrics.utilities.data import to_onehot from torchmetrics.utilities.enums import DataType, EnumStr -from torchmetrics.utilities.warn import _future_warning class MulticlassMode(EnumStr): diff --git a/torchmetrics/functional/classification/iou.py b/torchmetrics/functional/classification/iou.py index d4d634e8c4a..cba06f0d46d 100644 --- a/torchmetrics/functional/classification/iou.py +++ b/torchmetrics/functional/classification/iou.py @@ -18,7 +18,7 @@ from torch import Tensor from torchmetrics.functional.classification.jaccard import jaccard_index -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning @deprecated(target=jaccard_index, deprecated_in="0.7", remove_in="0.8", stream=_future_warning) diff --git a/torchmetrics/functional/image/ms_ssim.py b/torchmetrics/functional/image/ms_ssim.py index 850abeff12a..89100f88b52 100644 --- a/torchmetrics/functional/image/ms_ssim.py +++ b/torchmetrics/functional/image/ms_ssim.py @@ -19,7 +19,7 @@ from typing_extensions import Literal from torchmetrics.functional.image.ssim import multiscale_structural_similarity_index_measure as _ms_ssim -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning @deprecated(target=_ms_ssim, deprecated_in="0.7", remove_in="0.8", stream=_future_warning) diff --git a/torchmetrics/functional/image/psnr.py b/torchmetrics/functional/image/psnr.py index 78e6ab6609c..f99e5995b36 100644 --- a/torchmetrics/functional/image/psnr.py +++ b/torchmetrics/functional/image/psnr.py @@ -17,8 +17,7 @@ from deprecate import deprecated, void from torch import Tensor, tensor -from torchmetrics.utilities import rank_zero_warn, reduce -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning, rank_zero_warn, reduce def _psnr_compute( diff --git a/torchmetrics/functional/image/ssim.py b/torchmetrics/functional/image/ssim.py index b9eb2028a54..8c033d5e154 100644 --- a/torchmetrics/functional/image/ssim.py +++ b/torchmetrics/functional/image/ssim.py @@ -19,9 +19,9 @@ from torch.nn import functional as F from typing_extensions import Literal +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.checks import _check_same_shape from torchmetrics.utilities.distributed import reduce -from torchmetrics.utilities.warn import _future_warning def _gaussian(kernel_size: int, sigma: float, dtype: torch.dtype, device: torch.device) -> Tensor: diff --git a/torchmetrics/functional/pairwise/manhatten.py b/torchmetrics/functional/pairwise/manhatten.py index 34d8244ac37..a8634d57d24 100644 --- a/torchmetrics/functional/pairwise/manhatten.py +++ b/torchmetrics/functional/pairwise/manhatten.py @@ -17,7 +17,7 @@ from torch import Tensor from torchmetrics.functional.pairwise.manhattan import pairwise_manhattan_distance -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning @deprecated(target=pairwise_manhattan_distance, deprecated_in="0.7", remove_in="0.8", stream=_future_warning) diff --git a/torchmetrics/functional/text/bert.py b/torchmetrics/functional/text/bert.py index 4955fb64dd7..12d7372c481 100644 --- a/torchmetrics/functional/text/bert.py +++ b/torchmetrics/functional/text/bert.py @@ -23,8 +23,8 @@ from torch import Tensor from torch.utils.data import DataLoader, Dataset +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.imports import _TQDM_AVAILABLE, _TRANSFORMERS_AVAILABLE -from torchmetrics.utilities.warn import _future_warning if _TRANSFORMERS_AVAILABLE: from transformers import AutoModel, AutoTokenizer diff --git a/torchmetrics/functional/text/bleu.py b/torchmetrics/functional/text/bleu.py index 761c7266c86..880ef4ed7b8 100644 --- a/torchmetrics/functional/text/bleu.py +++ b/torchmetrics/functional/text/bleu.py @@ -24,7 +24,7 @@ from deprecate import deprecated from torch import Tensor, tensor -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning def _count_ngram(ngram_input_list: Sequence[str], n_gram: int) -> Counter: diff --git a/torchmetrics/functional/text/cer.py b/torchmetrics/functional/text/cer.py index 918af8dabf8..ba0bb47bfc4 100644 --- a/torchmetrics/functional/text/cer.py +++ b/torchmetrics/functional/text/cer.py @@ -19,7 +19,7 @@ from torch import Tensor, tensor from torchmetrics.functional.text.helper import _edit_distance -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning def _cer_update( diff --git a/torchmetrics/functional/text/sacre_bleu.py b/torchmetrics/functional/text/sacre_bleu.py index 6b1012aa349..730ddcbecad 100644 --- a/torchmetrics/functional/text/sacre_bleu.py +++ b/torchmetrics/functional/text/sacre_bleu.py @@ -48,8 +48,8 @@ from typing_extensions import Literal from torchmetrics.functional.text.bleu import _bleu_score_compute, _bleu_score_update +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.imports import _REGEX_AVAILABLE -from torchmetrics.utilities.warn import _future_warning AVAILABLE_TOKENIZERS = ("none", "13a", "zh", "intl", "char") diff --git a/torchmetrics/functional/text/wer.py b/torchmetrics/functional/text/wer.py index e977658bec3..093a96f68b7 100644 --- a/torchmetrics/functional/text/wer.py +++ b/torchmetrics/functional/text/wer.py @@ -19,7 +19,7 @@ from torch import Tensor, tensor from torchmetrics.functional.text.helper import _edit_distance -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning def _wer_update( diff --git a/torchmetrics/image/psnr.py b/torchmetrics/image/psnr.py index 5aa1d9fb562..cfacc6f364e 100644 --- a/torchmetrics/image/psnr.py +++ b/torchmetrics/image/psnr.py @@ -19,8 +19,7 @@ from torchmetrics.functional.image.psnr import _psnr_compute, _psnr_update from torchmetrics.metric import Metric -from torchmetrics.utilities import rank_zero_warn -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning, rank_zero_warn class PeakSignalNoiseRatio(Metric): diff --git a/torchmetrics/image/ssim.py b/torchmetrics/image/ssim.py index 86f482d9958..1110c59a074 100644 --- a/torchmetrics/image/ssim.py +++ b/torchmetrics/image/ssim.py @@ -20,9 +20,8 @@ from torchmetrics.functional.image.ssim import _multiscale_ssim_compute, _ssim_compute, _ssim_update from torchmetrics.metric import Metric -from torchmetrics.utilities import rank_zero_warn +from torchmetrics.utilities import _future_warning, rank_zero_warn from torchmetrics.utilities.data import dim_zero_cat -from torchmetrics.utilities.warn import _future_warning class StructuralSimilarityIndexMeasure(Metric): diff --git a/torchmetrics/regression/pearson.py b/torchmetrics/regression/pearson.py index 90b56a08ccd..d4a19259c1d 100644 --- a/torchmetrics/regression/pearson.py +++ b/torchmetrics/regression/pearson.py @@ -19,7 +19,7 @@ from torchmetrics.functional.regression.pearson import _pearson_corrcoef_compute, _pearson_corrcoef_update from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning def _final_aggregation( diff --git a/torchmetrics/regression/spearman.py b/torchmetrics/regression/spearman.py index 707328b41ac..549ce77c62f 100644 --- a/torchmetrics/regression/spearman.py +++ b/torchmetrics/regression/spearman.py @@ -19,9 +19,8 @@ from torchmetrics.functional.regression.spearman import _spearman_corrcoef_compute, _spearman_corrcoef_update from torchmetrics.metric import Metric -from torchmetrics.utilities import rank_zero_warn +from torchmetrics.utilities import _future_warning, rank_zero_warn from torchmetrics.utilities.data import dim_zero_cat -from torchmetrics.utilities.warn import _future_warning class SpearmanCorrCoef(Metric): diff --git a/torchmetrics/text/bert.py b/torchmetrics/text/bert.py index 432bff0893c..aec0c39e949 100644 --- a/torchmetrics/text/bert.py +++ b/torchmetrics/text/bert.py @@ -20,8 +20,8 @@ from torchmetrics.functional import bert_score from torchmetrics.functional.text.bert import _preprocess_text from torchmetrics.metric import Metric +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.imports import _TRANSFORMERS_AVAILABLE -from torchmetrics.utilities.warn import _future_warning if _TRANSFORMERS_AVAILABLE: from transformers import AutoTokenizer diff --git a/torchmetrics/text/bleu.py b/torchmetrics/text/bleu.py index 59992ecf77a..8d104a464e6 100644 --- a/torchmetrics/text/bleu.py +++ b/torchmetrics/text/bleu.py @@ -25,7 +25,7 @@ from torchmetrics import Metric from torchmetrics.functional.text.bleu import _bleu_score_compute, _bleu_score_update, _tokenize_fn -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class BLEUScore(Metric): diff --git a/torchmetrics/text/cer.py b/torchmetrics/text/cer.py index 5045611cf67..8771caf7dc4 100644 --- a/torchmetrics/text/cer.py +++ b/torchmetrics/text/cer.py @@ -20,7 +20,7 @@ from torchmetrics.functional.text.cer import _cer_compute, _cer_update from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class CharErrorRate(Metric): diff --git a/torchmetrics/text/sacre_bleu.py b/torchmetrics/text/sacre_bleu.py index bc4c646d4da..aca3a7b9055 100644 --- a/torchmetrics/text/sacre_bleu.py +++ b/torchmetrics/text/sacre_bleu.py @@ -26,8 +26,8 @@ from torchmetrics.functional.text.bleu import _bleu_score_update from torchmetrics.functional.text.sacre_bleu import _SacreBLEUTokenizer from torchmetrics.text.bleu import BLEUScore +from torchmetrics.utilities import _future_warning from torchmetrics.utilities.imports import _REGEX_AVAILABLE -from torchmetrics.utilities.warn import _future_warning AVAILABLE_TOKENIZERS = ("none", "13a", "zh", "intl", "char") diff --git a/torchmetrics/text/wer.py b/torchmetrics/text/wer.py index c1c74a1520f..ac0a012102d 100644 --- a/torchmetrics/text/wer.py +++ b/torchmetrics/text/wer.py @@ -19,7 +19,7 @@ from torchmetrics.functional.text.wer import _wer_compute, _wer_update from torchmetrics.metric import Metric -from torchmetrics.utilities.warn import _future_warning +from torchmetrics.utilities import _future_warning class WordErrorRate(Metric): diff --git a/torchmetrics/utilities/__init__.py b/torchmetrics/utilities/__init__.py index 00915078d2e..5049a9d6901 100644 --- a/torchmetrics/utilities/__init__.py +++ b/torchmetrics/utilities/__init__.py @@ -1,4 +1,3 @@ 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 rank_zero_debug, rank_zero_info, rank_zero_warn # noqa: F401 -from torchmetrics.utilities.warn import _future_warning # noqa: F401 +from torchmetrics.utilities.prints import _future_warning, rank_zero_debug, rank_zero_info, rank_zero_warn # noqa: F401 diff --git a/torchmetrics/utilities/prints.py b/torchmetrics/utilities/prints.py index 7fb783ed589..3188df0024c 100644 --- a/torchmetrics/utilities/prints.py +++ b/torchmetrics/utilities/prints.py @@ -13,9 +13,11 @@ # limitations under the License. import os import warnings -from functools import wraps +from functools import partial, wraps from typing import Any, Callable +from _warnings import warn + from torchmetrics import _logger as log @@ -47,3 +49,4 @@ def _debug(*args: Any, **kwargs: Any) -> None: rank_zero_debug = rank_zero_only(_debug) rank_zero_info = rank_zero_only(_info) rank_zero_warn = rank_zero_only(_warn) +_future_warning = partial(warn, category=FutureWarning) diff --git a/torchmetrics/utilities/warn.py b/torchmetrics/utilities/warn.py deleted file mode 100644 index ba68de12949..00000000000 --- a/torchmetrics/utilities/warn.py +++ /dev/null @@ -1,18 +0,0 @@ -# Copyright The PyTorch Lightning team. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# 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 functools import partial -from warnings import warn - -_future_warning = partial(warn, category=FutureWarning)