imports

.github/assistant.py

@@ -16,7 +16,7 @@
 import os
 import re
 import sys
-from typing import List, Optional, Tuple, Union
+from typing import Optional, Union
 
 import fire
 from packaging.version import parse

_samples/bert_score-own_model.py

@@ -18,7 +18,7 @@
 """
 
 from pprint import pprint
-from typing import Dict, List, Union
+from typing import Union
 
 import torch
 from torch import Tensor, nn

examples/audio/signal_to_noise_ratio.py

@@ -8,7 +8,6 @@
 
 # %%
 # Import necessary libraries
-from typing import Tuple
 
 import matplotlib.animation as animation
 import matplotlib.pyplot as plt

pyproject.toml

@@ -77,9 +77,6 @@ lint.per-file-ignores."tests/**" = [
   "S101",
   "S301",   # todo: `pickle` and modules that wrap it can be unsafe when used to deserialize untrusted data, possible security issue
 ]
-lint.unfixable = [
-  "F401",
-]
 # Unlike Flake8, default to a complexity level of 10.
 lint.mccabe.max-complexity = 10
 # Use Google-style docstrings.

setup.py

@@ -7,7 +7,7 @@
 from importlib.util import module_from_spec, spec_from_file_location
 from itertools import chain
 from pathlib import Path
-from typing import Any, List, Optional, Tuple, Union
+from typing import Any, Optional, Union
 
 from pkg_resources import Requirement, yield_lines
 from setuptools import find_packages, setup

src/torchmetrics/aggregation.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Callable, List, Optional, Tuple, Union
+from typing import Any, Callable, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/audio/pit.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Callable, Dict, Optional, Union
+from typing import Any, Callable, Optional, Union
 
 from torch import Tensor, tensor
 from typing_extensions import Literal

src/torchmetrics/classification/accuracy.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/auroc.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/average_precision.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/calibration_error.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/cohen_kappa.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/confusion_matrix.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 typing import Any, List, Optional, Type
+from typing import Any, Optional
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/dice.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Callable, Optional, Tuple, Union, no_type_check
+from typing import Any, Callable, Optional, Union, no_type_check
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/exact_match.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/f_beta.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/group_fairness.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Any, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/hamming.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/hinge.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/jaccard.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/matthews_corrcoef.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/negative_predictive_value.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/precision_fixed_recall.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/precision_recall.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/precision_recall_curve.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 typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/classification/recall_fixed_precision.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/roc.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 typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/sensitivity_specificity.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 typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/specificity.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, Optional, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/specificity_sensitivity.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 typing import Any, List, Optional, Tuple, Type, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 from typing_extensions import Literal

src/torchmetrics/classification/stat_scores.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 typing import Any, Callable, List, Optional, Tuple, Type, Union
+from typing import Any, Callable, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/clustering/adjusted_mutual_info_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Literal, Optional, Union
+from typing import Any, Literal, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/adjusted_rand_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/calinski_harabasz_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/davies_bouldin_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/dunn_index.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/fowlkes_mallows_index.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/homogeneity_completeness_v_measure.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/mutual_info_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/normalized_mutual_info_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Literal, Optional, Union
+from typing import Any, Literal, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/clustering/rand_score.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from torch import Tensor
 

src/torchmetrics/collections.py

@@ -15,7 +15,7 @@
 from collections import OrderedDict
 from collections.abc import Hashable, Iterable, Iterator, Mapping, Sequence
 from copy import deepcopy
-from typing import Any, ClassVar, Dict, List, Optional, Tuple, Union
+from typing import Any, ClassVar, Optional, Union
 
 import torch
 from torch import Tensor

src/torchmetrics/detection/_mean_ap.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import logging
 from collections.abc import Sequence
-from typing import Any, Callable, Dict, List, Optional, Tuple, Union
+from typing import Any, Callable, Optional, Union
 
 import numpy as np
 import torch

src/torchmetrics/detection/helpers.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 from collections.abc import Sequence
-from typing import Dict, Literal, Tuple, Union
+from typing import Literal, Union
 
 from torch import Tensor