Fix flexible lint toolchain drift

Author: Trinkle23897

BUILD

@@ -1,4 +1,5 @@
 load("@pip_requirements//:requirements.bzl", "requirement")
+load("@rules_python//python:defs.bzl", "py_binary")
 
 filegroup(
     name = "clang_tidy_config",

Makefile

@@ -53,10 +53,22 @@ addlicense-install: go-install
 doc-install:
 	$(call check_install, pydocstyle)
 	$(call check_install_extra, doc8, "doc8<1")
+	$(call check_install, setuptools)
+	$(call check_install, pbr)
 	$(call check_install, sphinx)
 	$(call check_install, sphinx_rtd_theme)
 	$(call check_install_extra, sphinxcontrib.spelling, sphinxcontrib.spelling pyenchant)
 
+spelling-system-install:
+	python3 -c "import ctypes.util, sys; sys.exit(0 if ctypes.util.find_library('enchant-2') or ctypes.util.find_library('enchant') else 1)" || \
+	([ "$$(uname -s)" = "Linux" ] && \
+	if command -v sudo >/dev/null 2>&1; then \
+	  sudo apt-get update && sudo apt-get install -y libenchant-2-dev; \
+	else \
+	  apt-get update && apt-get install -y libenchant-2-dev; \
+	fi && \
+	python3 -c "import ctypes.util, sys; sys.exit(0 if ctypes.util.find_library('enchant-2') or ctypes.util.find_library('enchant') else 1)")
+
 auditwheel-install:
 	$(call check_install_extra, auditwheel, auditwheel typed-ast patchelf)
 
@@ -119,15 +131,15 @@ bazel-clean: bazel-install
 # documentation
 
 addlicense: addlicense-install
-	addlicense -c $(COPYRIGHT) -l apache -y 2023 -check $(PROJECT_FOLDER)
+	addlicense -c $(COPYRIGHT) -l apache -y 2026 -check $(PROJECT_FOLDER)
 
 docstyle: doc-install
 	pydocstyle $(PROJECT_NAME) && doc8 docs && cd docs && make html SPHINXOPTS="-W"
 
 doc: doc-install
 	cd docs && make html && cd _build/html && python3 -m http.server
 
-spelling: doc-install
+spelling: doc-install spelling-system-install
 	cd docs && make spelling SPHINXOPTS="-W"
 
 doc-clean:
@@ -144,7 +156,7 @@ format: py-format-install clang-format-install buildifier-install addlicense-ins
 	yapf -ir $(PYTHON_FILES)
 	clang-format -style=file -i $(CPP_FILES)
 	buildifier -r -lint=fix $(BAZEL_FILES)
-	addlicense -c $(COPYRIGHT) -l apache -y 2023 $(PROJECT_FOLDER)
+	addlicense -c $(COPYRIGHT) -l apache -y 2026 $(PROJECT_FOLDER)
 
 # Build docker images
 

docs/conf.py

@@ -16,8 +16,6 @@
 
 import os
 
-import sphinx_rtd_theme
-
 
 def get_version() -> str:
   # https://packaging.python.org/guides/single-sourcing-package-version/
@@ -63,7 +61,6 @@ def get_version() -> str:
 # a list of builtin themes.
 #
 html_theme = "sphinx_rtd_theme"
-html_theme_path = [sphinx_rtd_theme.get_html_theme_path()]
 
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,

docs/spelling_wordlist.txt

@@ -13,6 +13,8 @@ bazel
 bazelisk
 buildifier
 addlicense
+isort
+yapf
 envpool
 th
 rew

envpool/BUILD

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 load("@pip_requirements//:requirements.bzl", "requirement")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 package(default_visibility = ["//visibility:public"])
 

envpool/atari/BUILD

@@ -14,6 +14,8 @@
 
 load("@pip_requirements//:requirements.bzl", "requirement")
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+load("@rules_cc//cc:defs.bzl", "cc_library", "cc_test")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 package(default_visibility = ["//visibility:public"])
 

envpool/atari/atari_network.py

@@ -45,15 +45,17 @@ def __init__(
       nn.Flatten(),
     )
     with torch.no_grad():
-      self.output_dim = np.prod(self.net(torch.zeros(1, c, h, w)).shape[1:])
+      self.output_dim = int(
+        np.prod(self.net(torch.zeros(1, c, h, w)).shape[1:])
+      )
     if not features_only:
       self.net = nn.Sequential(
         self.net,
         nn.Linear(self.output_dim, 512),
         nn.ReLU(inplace=True),
-        nn.Linear(512, np.prod(action_shape)),
+        nn.Linear(512, int(np.prod(action_shape))),
       )
-      self.output_dim = np.prod(action_shape)
+      self.output_dim = int(np.prod(action_shape))
 
   @no_type_check
   def forward(
@@ -80,7 +82,7 @@ def __init__(
     device: Union[str, int, torch.device] = "cpu",
   ) -> None:
     """Constructor of C51."""
-    self.action_num = np.prod(action_shape)
+    self.action_num = int(np.prod(action_shape))
     super().__init__(c, h, w, [self.action_num * num_atoms], device)
     self.num_atoms = num_atoms
 
@@ -111,7 +113,7 @@ def __init__(
     device: Union[str, int, torch.device] = "cpu",
   ) -> None:
     """Constructor of QRDQN."""
-    self.action_num = np.prod(action_shape)
+    self.action_num = int(np.prod(action_shape))
     super().__init__(c, h, w, [self.action_num * num_quantiles], device)
     self.num_quantiles = num_quantiles
 

envpool/box2d/BUILD

@@ -14,6 +14,8 @@
 
 load("@pip_requirements//:requirements.bzl", "requirement")
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+load("@rules_cc//cc:defs.bzl", "cc_library")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 package(default_visibility = ["//visibility:public"])
 

envpool/box2d/box2d_correctness_test.py

@@ -60,9 +60,7 @@ def test_car_racing_space(self) -> None:
     self.run_space_check(env0, env1)
 
   @staticmethod
-  def heuristic_lunar_lander_policy(
-    s: np.ndarray, continuous: bool
-  ) -> np.ndarray:
+  def heuristic_lunar_lander_policy(s: np.ndarray, continuous: bool) -> Any:
     angle_targ = np.clip(s[0] * 0.5 + s[2] * 1.0, -0.4, 0.4)
     hover_targ = 0.55 * np.abs(s[0])
     angle_todo = (angle_targ - s[4]) * 0.5 - s[5] * 1.0
@@ -73,17 +71,15 @@ def heuristic_lunar_lander_policy(
       hover_todo = -(s[3]) * 0.5
 
     if continuous:
-      a = np.array([hover_todo * 20 - 1, -angle_todo * 20])
-      a = np.clip(a, -1, 1)
-    else:
-      a = 0
-      if hover_todo > np.abs(angle_todo) and hover_todo > 0.05:
-        a = 2
-      elif angle_todo < -0.05:
-        a = 3
-      elif angle_todo > 0.05:
-        a = 1
-    return a
+      action = np.array([hover_todo * 20 - 1, -angle_todo * 20])
+      return np.clip(action, -1, 1)
+    if hover_todo > np.abs(angle_todo) and hover_todo > 0.05:
+      return 2
+    if angle_todo < -0.05:
+      return 3
+    if angle_todo > 0.05:
+      return 1
+    return 0
 
   def solve_lunar_lander(self, num_envs: int, continuous: bool) -> None:
     if continuous:

envpool/classic_control/BUILD

@@ -14,6 +14,8 @@
 
 load("@pip_requirements//:requirements.bzl", "requirement")
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
+load("@rules_cc//cc:defs.bzl", "cc_library")
+load("@rules_python//python:defs.bzl", "py_library", "py_test")
 
 package(default_visibility = ["//visibility:public"])