bump mypy to 0.981, enable recursive type aliases

.github/workflows/tests.yml

@@ -21,17 +21,7 @@ jobs:
       fail-fast: false
       matrix:
         os: [ubuntu-20.04, windows-latest, macos-latest]
-        pyv: ["3.8", "3.9"]
-        include:
-        - os: ubuntu-20.04
-          pyv: 3.10.6
-          nox_pyv: 3.10
-        - os: windows-latest
-          pyv: 3.10.6
-          nox_pyv: 3.10
-        - os: macos-latest
-          pyv: 3.10.6
-          nox_pyv: 3.10
+        pyv: ["3.8", "3.9", "3.10"]
 
     steps:
       - name: Check out the repository

pyproject.toml

@@ -63,6 +63,7 @@ warn_redundant_casts = true
 warn_unreachable = true
 ignore_missing_imports = true
 files = ["src", "tests"]
+enable_recursive_aliases = true
 
 [tool.pylint.message_control]
 enable = ["c-extension-no-member", "no-else-return"]

setup.cfg

@@ -39,7 +39,7 @@ tests =
     pytest-mock==3.8.2
     pylint==2.15.0
     pylint-plugin-utils>=0.6
-    mypy==0.971
+    mypy==0.981
 dev =
     %(tests)s
     %(all)s

src/dvclive/keras.py

@@ -1,5 +1,5 @@
 import os
-from typing import Optional
+from typing import Dict, Optional
 
 from tensorflow.keras.callbacks import (  # noqa pylint: disable=import-error, no-name-in-module
     Callback,
@@ -41,7 +41,7 @@ def on_train_begin(self, logs=None):  # pylint: disable=unused-argument
                 )
 
     def on_epoch_end(
-        self, epoch: int, logs: dict = None
+        self, epoch: int, logs: Optional[Dict] = None
     ):  # pylint: disable=unused-argument
         logs = logs or {}
         for metric, value in logs.items():

src/dvclive/live.py

@@ -27,11 +27,9 @@
 logger = logging.getLogger("dvclive")
 logger.setLevel(os.getenv(env.DVCLIVE_LOGLEVEL, "INFO").upper())
 
-
-# Recursive type aliases are not yet supported by mypy (as of 0.971),
-# so we set type: ignore for ParamLike.
-#  See https://github.com/python/mypy/issues/731#issuecomment-1213482527
-ParamLike = Union[int, float, str, bool, List["ParamLike"], Dict[str, "ParamLike"]]  # type: ignore # noqa
+ParamLike = Union[
+    int, float, str, bool, List["ParamLike"], Dict[str, "ParamLike"]
+]
 
 
 class Live: