[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

modify_changelog.py

@@ -77,7 +77,7 @@ def find_unreleased_section(lines):
 
 def update_changelog_version(file_path: Path, new_version: str) -> None:
     """Update changelog for version bump - replaces UNRELEASED with versioned section."""
-    today = datetime.datetime.now(tz=datetime.timezone.utc).strftime("%Y-%m-%d")
+    today = datetime.datetime.now(tz=datetime.UTC).strftime("%Y-%m-%d")
     content = file_path.read_text()
     new_section = f"## UNRELEASED\n\n## {new_version} ({today})"
     content = content.replace("## UNRELEASED", new_section, 1)

python/egglog/bindings.pyi

@@ -2,9 +2,7 @@ from collections.abc import Callable
 from datetime import timedelta
 from fractions import Fraction
 from pathlib import Path
-from typing import Any, Generic, Protocol, TypeAlias, TypeVar
-
-from typing_extensions import final
+from typing import Any, Generic, Protocol, TypeAlias, TypeVar, final
 
 __all__ = [
     "ActionCommand",

python/egglog/declarations.py

@@ -8,12 +8,22 @@
 
 from dataclasses import dataclass, field
 from functools import cached_property
-from typing import TYPE_CHECKING, ClassVar, Literal, Protocol, TypeAlias, TypeVar, Union, cast, runtime_checkable
+from typing import (
+    TYPE_CHECKING,
+    ClassVar,
+    Literal,
+    Protocol,
+    Self,
+    TypeAlias,
+    TypeVar,
+    Union,
+    assert_never,
+    cast,
+    runtime_checkable,
+)
 from uuid import UUID
 from weakref import WeakValueDictionary
 
-from typing_extensions import Self, assert_never
-
 from .bindings import Value
 
 if TYPE_CHECKING:

python/egglog/deconstruct.py

@@ -47,7 +47,7 @@ def get_literal_value(x: PyObject) -> object: ...
 
 
 @overload
-def get_literal_value(x: UnstableFn[T, Unpack[TS]]) -> Callable[[Unpack[TS]], T] | None: ...
+def get_literal_value(x: UnstableFn[T, *TS]) -> Callable[[Unpack[TS]], T] | None: ...
 
 
 @overload
@@ -118,10 +118,10 @@ def get_callable_args(x: T, fn: None = ...) -> tuple[BaseExpr, ...]: ...
 
 
 @overload
-def get_callable_args(x: T, fn: Callable[[Unpack[TS]], T]) -> tuple[Unpack[TS]] | None: ...
+def get_callable_args(x: T, fn: Callable[[Unpack[TS]], T]) -> tuple[*TS] | None: ...
 
 
-def get_callable_args(x: T, fn: Callable[[Unpack[TS]], T] | None = None) -> tuple[Unpack[TS]] | None:
+def get_callable_args(x: T, fn: Callable[[Unpack[TS]], T] | None = None) -> tuple[*TS] | None:
     """
     Gets all the arguments of an expression.
     If a function is provided, it will only return the arguments if the expression is a call

python/egglog/egraph_state.py

@@ -8,11 +8,10 @@
 from base64 import standard_b64decode, standard_b64encode
 from collections import defaultdict
 from dataclasses import dataclass, field, replace
-from typing import TYPE_CHECKING, Literal, overload
+from typing import TYPE_CHECKING, Literal, assert_never, overload
 from uuid import UUID
 
 import cloudpickle
-from typing_extensions import assert_never
 
 from . import bindings
 from .declarations import *

python/egglog/thunk.py

@@ -33,15 +33,15 @@ def right(self) -> V:
 
 
 @dataclass
-class Thunk(Generic[T, Unpack[TS]]):
+class Thunk(Generic[T, *TS]):
     """
     Cached delayed function call.
     """
 
-    state: Resolved[T] | Unresolved[T, Unpack[TS]] | Resolving | Error
+    state: Resolved[T] | Unresolved[T, *TS] | Resolving | Error
 
     @classmethod
-    def fn(cls, fn: Callable[[Unpack[TS]], T], *args: Unpack[TS], context: str | None = None) -> Thunk[T, Unpack[TS]]:
+    def fn(cls, fn: Callable[[Unpack[TS]], T], *args: *TS, context: str | None = None) -> Thunk[T, *TS]:
         """
         Create a thunk based on some functions and some partial args.
 
@@ -80,9 +80,9 @@ class Resolved(Generic[T]):
 
 
 @dataclass
-class Unresolved(Generic[T, Unpack[TS]]):
+class Unresolved(Generic[T, *TS]):
     fn: Callable[[Unpack[TS]], T]
-    args: tuple[Unpack[TS]]
+    args: tuple[*TS]
     context: str | None
 
 

python/egglog/type_constraint_solver.py

@@ -5,9 +5,7 @@
 from collections import defaultdict
 from dataclasses import dataclass, field
 from itertools import chain, repeat
-from typing import TYPE_CHECKING
-
-from typing_extensions import assert_never
+from typing import TYPE_CHECKING, assert_never
 
 from .declarations import *