typing

PanopticQuality.py

@@ -12,9 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """TODO: Add a description here."""
-
-from typing import Set
-from deprecated import deprecated
+from typing import Dict, List
 
 import evaluate
 import datasets
@@ -80,8 +78,8 @@ Examples:
 class PQMetric(evaluate.Metric):
     def __init__(
         self,
-        label2id: dict = None,
-        stuff: list = None,
+        label2id: Dict[str, int] = None,
+        stuff: List[str] = None,
         **kwargs
     ):
         super().__init__(**kwargs)