Pre-commit

Author: pseudo-rnd-thoughts

.pre-commit-config.yaml

@@ -1,12 +1,61 @@
+# See https://pre-commit.com for more information
+# See https://pre-commit.com/hooks.html for more hooks
 repos:
--   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v2.3.0
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
     hooks:
-    -   id: check-yaml
-    -   id: end-of-file-fixer
-    -   id: trailing-whitespace
-    -   id: debug-statements
--   repo: https://github.com/psf/black
-    rev: 22.3.0
+      - id: check-symlinks
+      - id: destroyed-symlinks
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-toml
+      - id: check-ast
+      - id: check-added-large-files
+      - id: check-merge-conflict
+      - id: check-executables-have-shebangs
+      - id: check-shebang-scripts-are-executable
+      - id: detect-private-key
+      - id: debug-statements
+#  - repo: https://github.com/codespell-project/codespell
+#    rev: v2.2.6
+#    exclude: ^(src/common)|(src/emucore)|(src/environment)|(src/games)
+#    hooks:
+#      - id: codespell
+#        args:
+#          - --ignore-words-list=
+  - repo: https://github.com/PyCQA/flake8
+    rev: 7.0.0
     hooks:
-    -   id: black
+      - id: flake8
+        args:
+#          - '--per-file-ignores='
+          - --ignore=E203,W503,E741
+          - --max-complexity=30
+          - --max-line-length=456
+          - --show-source
+          - --statistics
+  - repo: https://github.com/asottile/pyupgrade
+    rev: v3.15.0
+    hooks:
+      - id: pyupgrade
+        args: ["--py38-plus"]
+  - repo: https://github.com/PyCQA/isort
+    rev: 5.13.2
+    hooks:
+      - id: isort
+        args: ["--profile", "black"]
+  - repo: https://github.com/python/black
+    rev: 23.12.1
+    hooks:
+      - id: black
+#  - repo: https://github.com/pycqa/pydocstyle
+#    rev: 6.3.0
+#    hooks:
+#      - id: pydocstyle
+##        exclude: ^
+#        args:
+#          - --source
+#          - --explain
+#          - --convention=google
+#        additional_dependencies: ["tomli"]

examples/python-interface/python_example.py

@@ -6,7 +6,8 @@
 # ALE provided in examples/sharedLibraryInterfaceExample.cpp
 import sys
 from random import randrange
-from ale_py import ALEInterface, SDL_SUPPORT
+
+from ale_py import SDL_SUPPORT, ALEInterface
 
 if len(sys.argv) < 2:
     print(f"Usage: {sys.argv[0]} rom_file")

examples/python-interface/python_example_with_modes.py

@@ -6,7 +6,8 @@
 # ALE provided in doc/examples/sharedLibraryInterfaceWithModesExample.cpp
 import sys
 from random import randrange
-from ale_py import ALEInterface, SDL_SUPPORT
+
+from ale_py import SDL_SUPPORT, ALEInterface
 
 if len(sys.argv) < 2:
     print(f"Usage: {sys.argv[0]} rom_file")
@@ -40,7 +41,6 @@
 # Play one episode in each mode and in each difficulty
 for mode in avail_modes:
     for diff in avail_diff:
-
         ale.setDifficulty(diff)
         ale.setMode(mode)
         ale.reset_game()

examples/python-rom-package/roms/__init__.py

@@ -1,5 +1,5 @@
-import sys
 import pathlib
+import sys
 
 if sys.version_info >= (3, 9):
     import importlib.resources as resources

setup.py

@@ -123,7 +123,7 @@ def parse_version(version_file):
     semver_regex = r"(?P<major>0|[1-9]\d*)\.(?P<minor>0|[1-9]\d*)\.(?P<patch>0|[1-9]\d*)(?:-(?P<prerelease>(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\.(?:0|[1-9]\d*|\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?(?:\+(?P<buildmetadata>[0-9a-zA-Z-]+(?:\.[0-9a-zA-Z-]+)*))?"
     semver_prog = re.compile(semver_regex)
 
-    with open(version_file, "r") as fp:
+    with open(version_file) as fp:
         version = fp.read().strip()
         assert semver_prog.match(version) is not None
 

src/python/__init__.py

@@ -45,14 +45,18 @@
     __version__ = "unknown"
 
 # Import native shared library
-from ale_py._ale_py import SDL_SUPPORT, Action, ALEInterface, ALEState, LoggerMode
+from ale_py._ale_py import (  # noqa: E402
+    SDL_SUPPORT,
+    Action,
+    ALEInterface,
+    ALEState,
+    LoggerMode,
+)
 
 __all__ = ["Action", "ALEInterface", "ALEState", "LoggerMode", "SDL_SUPPORT"]
 
 
 try:
-    import gymnasium
-
     from ale_py.env import AtariEnv, AtariEnvStepMetadata
 
     __all__ += ["AtariEnv", "AtariEnvStepMetadata"]

src/python/env.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import sys
-from typing import Any, Literal, Optional, Union
+from typing import Any, Literal
 
 import ale_py
 import gymnasium
@@ -35,14 +35,14 @@ class AtariEnv(gymnasium.Env, utils.EzPickle):
     def __init__(
         self,
         game: str,
-        mode: Optional[int] = None,
-        difficulty: Optional[int] = None,
+        mode: int | None = None,
+        difficulty: int | None = None,
         obs_type: Literal["rgb", "grayscale", "ram"] = "rgb",
-        frameskip: Union[tuple[int, int], int] = 4,
+        frameskip: tuple[int, int] | int = 4,
         repeat_action_probability: float = 0.25,
         full_action_space: bool = False,
-        max_num_frames_per_episode: Optional[int] = None,
-        render_mode: Optional[Literal["human", "rgb_array"]] = None,
+        max_num_frames_per_episode: int | None = None,
+        render_mode: Literal["human", "rgb_array"] | None = None,
     ):
         """
         Initialize the ALE for Gymnasium.
@@ -96,11 +96,11 @@ def __init__(
             )
         elif isinstance(frameskip, tuple) and frameskip[0] > frameskip[1]:
             raise error.Error(
-                f"Invalid stochastic frameskip, lower bound is greater than upper bound."
+                "Invalid stochastic frameskip, lower bound is greater than upper bound."
             )
         elif isinstance(frameskip, tuple) and frameskip[0] <= 0:
             raise error.Error(
-                f"Invalid stochastic frameskip lower bound is greater than upper bound."
+                "Invalid stochastic frameskip lower bound is greater than upper bound."
             )
 
         if render_mode is not None and render_mode not in {"rgb_array", "human"}:
@@ -176,7 +176,7 @@ def __init__(
         else:
             raise error.Error(f"Unrecognized observation type: {self._obs_type}")
 
-    def seed_game(self, seed: Optional[int] = None) -> tuple[int, int]:
+    def seed_game(self, seed: int | None = None) -> tuple[int, int]:
         """Seeds the internal and ALE RNG."""
         ss = np.random.SeedSequence(seed)
         np_seed, ale_seed = ss.generate_state(n_words=2)
@@ -196,8 +196,8 @@ def load_game(self) -> None:
     def reset(  # pyright: ignore[reportIncompatibleMethodOverride]
         self,
         *,
-        seed: Optional[int] = None,
-        options: Optional[dict[str, Any]] = None,
+        seed: int | None = None,
+        options: dict[str, Any] | None = None,
     ) -> tuple[np.ndarray, AtariEnvStepMetadata]:
         """Resets environment and returns initial observation."""
         super().reset(seed=seed, options=options)
@@ -253,7 +253,7 @@ def step(  # pyright: ignore[reportIncompatibleMethodOverride]
 
         return self._get_obs(), reward, is_terminal, is_truncated, self._get_info()
 
-    def render(self) -> Optional[np.ndarray]:
+    def render(self) -> np.ndarray | None:
         """
         Render is not supported by ALE. We use a paradigm similar to
         Gym3 which allows you to specify `render_mode` during construction.

src/python/registration.py

@@ -1,22 +1,21 @@
 from __future__ import annotations
 
 from collections import defaultdict
-from typing import Any, Callable, Mapping, NamedTuple, Sequence, Text, Union
+from typing import Any, Callable, Mapping, NamedTuple, Sequence
 
 import ale_py.roms as roms
-from ale_py.roms import utils as rom_utils
-
 import gymnasium
+from ale_py.roms import utils as rom_utils
 
 
 class EnvFlavour(NamedTuple):
     suffix: str
-    kwargs: Union[Mapping[Text, Any], Callable[[str], Mapping[Text, Any]]]
+    kwargs: Mapping[str, Any] | Callable[[str], Mapping[str, Any]]
 
 
 class EnvConfig(NamedTuple):
     version: str
-    kwargs: Mapping[Text, Any]
+    kwargs: Mapping[str, Any]
     flavours: Sequence[EnvFlavour]
 
 

src/python/roms/__init__.py

@@ -97,7 +97,7 @@ def __dir__() -> List[str]:
     md5s = resources.files(__name__).joinpath("md5.txt")
     if not md5s.exists():
         raise FileNotFoundError(
-            f"ROM md5 resource couldn't be found. "
+            "ROM md5 resource couldn't be found. "
             "Are you running from a development environment? "
         )
     with md5s.open() as fp:

tests/conftest.py

@@ -1,4 +1,5 @@
 import os
+
 import pytest
 
 
@@ -13,6 +14,3 @@ def __getitem__(self, file_name):
 @pytest.fixture(scope="module")
 def resources():
     return Resources(os.path.abspath(os.path.dirname(__file__)))
-
-
-from fixtures import *