⬆️ Upgrade dependencies and apply new formatting

.github/workflows/check.yml

@@ -12,7 +12,7 @@ jobs:
             fail-fast: false
             matrix:
                 os: [ ubuntu-latest, windows-latest, macos-latest ]
-                python: [ "3.9", "3.10", "3.11" ]
+                python: [ "3.10", "3.11", "3.12" ]
         uses: ./.github/workflows/setup.yml
         with:
             os: ${{ matrix.os }}

env.yml

@@ -2,4 +2,4 @@ name: safecheck
 channels:
   - conda-forge
 dependencies:
-  - conda-forge::python=3.9
+  - conda-forge::python=3.10

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "safecheck"
-version = "0.3.1"
+version = "0.4.0"
 description = "Utilities for typechecking, shapechecking and dispatch."
 readme = "README.md"
 authors = ["David Muhr <[email protected]>"]
@@ -38,39 +38,39 @@ classifiers = [
 ]
 
 [tool.poetry.dependencies]
-python = ">=3.9,<3.13"
-beartype = { version = "^0.16" }
-jaxtyping = { version = "^0.2.24" }
+python = ">=3.10,<3.13"
+beartype = { version = ">0.15 <0.19" }
+jaxtyping = { version = "^0.2" }
 typing-extensions = { version = "^4.0" }
 
 [tool.poetry.group.dev.dependencies]
 # testing
-pytest = "^7.4.3"
+pytest = "^8.3.3"
 pytest-html = "^4.1.1"
 pytest-benchmark = "^4.0.0"
-hypothesis = "^6.92.2"
+hypothesis = "^6.112.1"
 # coverage
-coverage = "^7.4.0"
-pytest-cov = "^4.1.0"
-coverage-badge = "^1.1.0"
+coverage = "^7.6.1"
+pytest-cov = "^5.0.0"
+coverage-badge = "^1.1.2"
 # linting
-ruff = "^0.1.9"
-pre-commit = "^3.6.0"
-black = "^23.12.1"
+ruff = "^0.6.5"
+pre-commit = "^3.8.0"
+black = "^24.8.0"
 # typing
-pyright = "^1.1.344"
-typing-extensions = "^4.9.0"
+pyright = "^1.1.380"
+typing-extensions = "^4.12.2"
 # safety
-bandit = "^1.7.6"
-safety = "^2.3.5"
+bandit = "^1.7.9"
+safety = "^3.2.7"
 # test data
-numpy = "^1.26.2"
+numpy = "^2.1.1"
 torch = [
-    { version = "^2.1.2", platform = "darwin", source = "pypi" },
-    { version = "^2.1.2", platform = "win32", source = "pypi" },
-    { version = "^2.1.2", platform = "linux", source = "pytorch-cpu" }
+    { version = "^2.4.1", platform = "darwin", source = "pypi" },
+    { version = "^2.4.1", platform = "win32", source = "pypi" },
+    { version = "^2.4.1", platform = "linux", source = "pytorch-cpu" }
 ]
-jax = { version = "^0.4.23", extras = ["cpu"] }
+jax = { version = "^0.4.33", extras = ["cpu"] }
 # notebooks
 notebook = "^7.0.4"
 # profiling
@@ -104,25 +104,25 @@ exclude = '''
 
 [tool.ruff]
 line-length = 120
-select = ["ALL"]
 extend-exclude = ["tests/", ".github/"]
 force-exclude = true
-ignore = [
+lint.select = ["ALL"]
+lint.ignore = [
     "D203", # one blank line required before class docstring
     "D213", # multi line summary should start at second line
     "ANN101", # missing type annotation for `self` in method
     "B905", # `zip()` without an explicit `strict=` parameter
 ]
 
-[tool.ruff.isort]
+[tool.ruff.lint.isort]
 # Combine multiple `from foo import bar as baz` statements with the same source
 # (`foo`) into a single statement.
 combine-as-imports = true
 # Imports of the form `from foo import bar as baz` show one `import bar as baz`
 # per line. Useful for __init__.py files that just re-export symbols.
 force-wrap-aliases = true
 
-[tool.ruff.flake8-quotes]
+[tool.ruff.lint.flake8-quotes]
 docstring-quotes = "double"
 
 [tool.pyright]

safecheck/__init__.py

@@ -6,6 +6,7 @@
 We additionally rename some of the functions to be independent of the underlying packages. For example,
 it should be easily possible to switch from beartype to typeguard for runtime type checking.
 """
+
 from warnings import filterwarnings
 
 # re-export everything necessary from beartype, never use beartype itself.

safecheck/_overload.py

@@ -14,6 +14,7 @@
 2. The overloads are iterated in definition-order until a matching overload is found.
 3. If a matching overload is found (typecheck successful) return the result, otherwise raise an error.
 """
+
 import sys
 import warnings
 from functools import wraps
@@ -32,16 +33,13 @@
 ]
 
 
-class UnavailableOverloadError(Exception):
-    ...
+class UnavailableOverloadError(Exception): ...
 
 
-class MissingOverloadError(Exception):
-    ...
+class MissingOverloadError(Exception): ...
 
 
-class IncompatibleOverloadError(Exception):
-    ...
+class IncompatibleOverloadError(Exception): ...
 
 
 def typecheck_overload(fn: CallableAnyT) -> CallableAnyT:

safecheck/_typecheck.py

@@ -9,8 +9,7 @@
 ]
 
 
-class MissingAnnotationError(Exception):
-    ...
+class MissingAnnotationError(Exception): ...
 
 
 CallableAnyT = Callable[..., Any]

tests/test_benchmark.py

@@ -13,30 +13,24 @@ def decorate(f):
     return f
 
 
-def f_basic(*_: int) -> None:
-    ...
+def f_basic(*_: int) -> None: ...
 
 
-def f_shaped(*_: Shaped[NumpyArray, "n"]) -> None:
-    ...
+def f_shaped(*_: Shaped[NumpyArray, "n"]) -> None: ...
 
 
 @overload
-def f_overload(*_: int) -> None:
-    ...
+def f_overload(*_: int) -> None: ...
 
 
-def f_overload(*_):
-    ...
+def f_overload(*_): ...
 
 
 @overload
-def f_overload_shaped(*_: Shaped[NumpyArray, "n"]) -> None:
-    ...
+def f_overload_shaped(*_: Shaped[NumpyArray, "n"]) -> None: ...
 
 
-def f_overload_shaped(*_):
-    ...
+def f_overload_shaped(*_): ...
 
 
 def test_no_overhead(benchmark):

tests/test_safecheck.py

@@ -102,8 +102,7 @@ def f(self, x: basic_type) -> basic_type:
 @pytest.mark.parametrize("basic_type", basic_types.keys())
 def test_basic_type_overload_function(basic_type):
     @overload
-    def f(x: basic_type) -> basic_type:
-        ...
+    def f(x: basic_type) -> basic_type: ...
 
     @typecheck_overload
     def f(x):
@@ -116,8 +115,7 @@ def f(x):
 def test_basic_type_overload_method(basic_type):
     class A:
         @overload
-        def f(self, x: basic_type) -> basic_type:
-            ...
+        def f(self, x: basic_type) -> basic_type: ...
 
         @typecheck_overload
         def f(self, x):
@@ -138,8 +136,7 @@ def f(x: type_to_check) -> type_to_check:
 @pytest.mark.parametrize("type_to_check", [union_type, generic_type])
 def test_union_type_overload_function(type_to_check):
     @overload
-    def f(x: type_to_check) -> type_to_check:
-        ...
+    def f(x: type_to_check) -> type_to_check: ...
 
     def f(x):
         return x
@@ -151,8 +148,7 @@ def f(x):
 def test_union_type_plain_method(type_to_check):
     class A:
         @overload
-        def f(self, x: type_to_check) -> type_to_check:
-            ...
+        def f(self, x: type_to_check) -> type_to_check: ...
 
         @typecheck_overload
         def f(self, x):
@@ -185,8 +181,7 @@ def f(array: data_type[array_type, "..."]) -> data_type[array_type, "..."]:
 @pytest.mark.parametrize("data_type", next(iter(data_types.values())).keys())
 def test_array_type_overload_function(array_type, data_type):
     @overload
-    def f(array: data_type[array_type, "..."]) -> data_type[array_type, "..."]:
-        ...
+    def f(array: data_type[array_type, "..."]) -> data_type[array_type, "..."]: ...
 
     @typecheck_overload
     def f(array):
@@ -211,8 +206,7 @@ def f(self, array: data_type[array_type, "..."]) -> data_type[array_type, "..."]
 def test_array_type_overload_method(array_type, data_type):
     class A:
         @overload
-        def f(self, array: data_type[array_type, "..."]) -> data_type[array_type, "..."]:
-            ...
+        def f(self, array: data_type[array_type, "..."]) -> data_type[array_type, "..."]: ...
 
         @typecheck_overload
         def f(self, array):
@@ -230,8 +224,7 @@ def test_missing_annotation():
 
 
 def test_missing_overload():
-    def f():
-        ...
+    def f(): ...
 
     class o:
         __annotations__ = {}
@@ -245,20 +238,17 @@ class o:
 
 def test_incompatible_overload():
     @overload
-    def f(a, b):
-        ...
+    def f(a, b): ...
 
-    def f(a):
-        ...
+    def f(a): ...
 
     with pytest.raises(IncompatibleOverloadError):
         typecheck_overload(f)
 
 
 def test_warn_overload_annotation():
     @overload
-    def f(x: int) -> int:
-        ...
+    def f(x: int) -> int: ...
 
     def f(x: float) -> float:
         return x
@@ -269,8 +259,7 @@ def f(x: float) -> float:
 
 def test_unavailable_overload():
     @overload
-    def f(x: int) -> int:
-        ...
+    def f(x: int) -> int: ...
 
     def f(x):
         return x