typing

Lightning-AI · Oct 6, 2022 · aae424f · aae424f
1 parent 2784439
commit aae424f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/torchmetrics/metric.py b/src/torchmetrics/metric.py
@@ -16,7 +16,7 @@
 from abc import ABC, abstractmethod
 from contextlib import contextmanager
 from copy import deepcopy
-from typing import Any, Callable, Dict, Generator, List, Optional, Sequence, Union
+from typing import Any, Callable, Dict, Generator, List, Optional, Sequence, Tuple, Union
 
 import torch
 from torch import Tensor
@@ -848,7 +848,7 @@ def __pos__(self) -> "Metric":
     def __getitem__(self, idx: int) -> "Metric":
         return CompositionalMetric(lambda x: x[idx], self, None)
 
-    def __getnewargs__(self):
+    def __getnewargs__(self) -> Tuple:
         return (Metric.__str__(self),)