Allow type checking on Python 3.11 (#1181)

Author: tmke8

ethicml/implementations/beutel.py

@@ -366,7 +366,7 @@ def forward(ctx: Any, x: Tensor, lambda_: float) -> Any:
         return x.view_as(x)
 
     @staticmethod
-    def backward(ctx: Any, grad_output: Tensor) -> Any:  # type: ignore[override]
+    def backward(ctx: Any, grad_output: Tensor) -> Any:  # pyright: ignore
         """Backward pass with Gradient reversed / inverted."""
         return grad_output.neg().mul(ctx.lambda_), None
 

ethicml/metrics/dependence_measures.py

@@ -156,4 +156,4 @@ def _corr(x: np.ndarray, y: np.ndarray) -> float:
 
 def _count_true(mask: np.ndarray) -> int:
     """Count the number of elements that are True."""
-    return np.count_nonzero(mask)
+    return np.count_nonzero(mask).item()

pyproject.toml

@@ -75,6 +75,7 @@ typecheck = [
     "pandas-stubs>=1.4.2.220626",
     "scipy-stubs>=1.15.3.0",
     "types-seaborn<1.0.0.0,>=0.13.2.20240205",
+    "backports.strenum>=1.3.1",
 ]
 lint = [
     "ruff>=0.2.2",