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

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

Author: pre-commit-ci[bot]

python/egglog/bindings.pyi

@@ -181,7 +181,6 @@ class Value:
     def __ge__(self, other: object) -> bool: ...
 
 @final
-
 @final
 class EggSmolError(Exception):
     context: str

python/egglog/builtins.py

@@ -499,7 +499,6 @@ def match(self, f: Callable[[T], V], n: V) -> V: ...
     # def flat_map(self, f: Callable[[T], Maybe[V]]) -> Maybe[V]: ...
 
 
-
 converter(type(None), Maybe, lambda _: Maybe[get_type_args()[0]].none())
 # converter(object, Maybe, lambda x: Maybe[get_type_args()[0]].some(convert(x, get_type_args()[0])))
 
@@ -719,7 +718,9 @@ def map_best_common_float_scale(xs: Map[T, f64]) -> f64: ...
 
 
 @function(egg_fn="map-integer-residual-split-candidate", builtin=True)
-def map_integer_residual_split_candidate(xs: Map[Map[T, BigRat], f64]) -> Pair[Map[T, BigRat], Map[Map[T, BigRat], f64]]: ...
+def map_integer_residual_split_candidate(
+    xs: Map[Map[T, BigRat], f64],
+) -> Pair[Map[T, BigRat], Map[Map[T, BigRat], f64]]: ...
 
 
 @function(egg_fn="map-factor-coef-for-integer-residual-split", builtin=True)

python/egglog/deconstruct.py

@@ -196,6 +196,7 @@ def _deconstruct_call_decl(
 
     return RuntimeFunction(decls_thunk, Thunk.value(call.callable), egg_bound), arg_exprs
 
+
 def is_expr_instance(x: BaseExpr, cls: type[T]) -> TypeIs[T]:
     """
     Checks if the expression is an instance of the given class. Can normally use isinstance for this, but this also works for

python/egglog/egraph.py

@@ -142,7 +142,9 @@
 }
 
 
-def check_eq(x: BASE_EXPR, y: BASE_EXPR, schedule: Schedule | None = None, *actions: ActionLike, display=False) -> EGraph:
+def check_eq(
+    x: BASE_EXPR, y: BASE_EXPR, schedule: Schedule | None = None, *actions: ActionLike, display=False
+) -> EGraph:
     """
     Verifies that two expressions are equal after running the schedule.
 
@@ -478,9 +480,7 @@ def _generate_class_decls(  # noqa: C901,PLR0912
                 subsume=False,
             )
             resolved_default = (
-                resolve_literal(type_ref, default_value, Thunk.value(decls))
-                if default_mode == "eager"
-                else None
+                resolve_literal(type_ref, default_value, Thunk.value(decls)) if default_mode == "eager" else None
             )
             if resolved_default is not None:
                 decls |= resolved_default
@@ -911,11 +911,7 @@ def _constant_thunk(
         unextractable=False,
         subsume=False,
     )
-    resolved_default = (
-        resolve_literal(type_ref, default_replacement, Thunk.value(decls))
-        if mode == "eager"
-        else None
-    )
+    resolved_default = resolve_literal(type_ref, default_replacement, Thunk.value(decls)) if mode == "eager" else None
     if resolved_default is not None:
         decls |= resolved_default
     decls._constants[ident] = ConstantDecl(
@@ -1596,11 +1592,7 @@ def all_function_sizes(self) -> list[tuple[ExprCallable, int]]:
         """
         (output,) = self._state.run_program(bindings.PrintSize(span(1), None))
         assert isinstance(output, bindings.PrintAllFunctionsSize)
-        return [
-            (callables[0], size)
-            for (name, size) in output.sizes
-            if (callables := self._egg_fn_to_callables(name))
-        ]
+        return [(callables[0], size) for (name, size) in output.sizes if (callables := self._egg_fn_to_callables(name))]
 
     def _egg_fn_to_callables(self, egg_fn: str) -> list[ExprCallable]:
         return [

python/egglog/egraph_state.py

@@ -410,7 +410,7 @@ def ruleset_to_egg(self, ident: Ident) -> None:
     def command_to_egg(self, cmd: CommandDecl, ruleset: Ident) -> bindings._Command | None:
         match cmd:
             case ActionCommandDecl(action):
-                action_egg = self.action_to_egg(action)# , expr_to_let=False)
+                action_egg = self.action_to_egg(action)  # , expr_to_let=False)
                 if not action_egg:
                     return None
                 return bindings.ActionCommand(action_egg)
@@ -641,9 +641,9 @@ def _primitive_command_to_egg(
         signature: FunctionSignature,
         body: TypedExprDecl,
     ) -> bindings.UserDefined:
-        input_sort_expr = self._primitive_input_sorts_to_egg(
-            [self.type_ref_to_egg(arg_type.to_just()) for arg_type in signature.arg_types]
-        )
+        input_sort_expr = self._primitive_input_sorts_to_egg([
+            self.type_ref_to_egg(arg_type.to_just()) for arg_type in signature.arg_types
+        ])
         output_sort_expr = bindings.Var(span(), self.type_ref_to_egg(signature.semantic_return_type.to_just()))
         return bindings.UserDefined(
             span(),

python/egglog/exp/__snapshots__/test_polynomials/test_factor_multisets[code].py

@@ -1,16 +1,47 @@
-_expr_1 = Value.var("q2") * Value.var("bp1") + Value.var("q5") * Value.var("bp2") + Value.var("q8") * Value.var("bp3") + Value.var("bp4") * Value.var("q11")
-_expr_2 = Value.var("bpp2") * Value.var("q6") + Value.var("q3") * Value.var("bpp1") + Value.var("bpp3") * Value.var("q9") + Value.var("bpp4") * Value.var("q12")
-_expr_3 = Value.var("bp1") * Value.var("q3") + Value.var("bp2") * Value.var("q6") + Value.var("q12") * Value.var("bp4") + Value.var("bp3") * Value.var("q9")
-_expr_4 = Value.var("bpp2") * Value.var("q5") + Value.var("q2") * Value.var("bpp1") + Value.var("bpp3") * Value.var("q8") + Value.var("bpp4") * Value.var("q11")
-_expr_5 = Value.var("q4") * Value.var("bpp2") + Value.var("q7") * Value.var("bpp3") + Value.var("q10") * Value.var("bpp4") + Value.var("bpp1") * Value.var("q1")
-_expr_6 = Value.var("q4") * Value.var("bp2") + Value.var("q7") * Value.var("bp3") + Value.var("bp1") * Value.var("q1") + Value.var("q10") * Value.var("bp4")
+_expr_1 = (
+    Value.var("q2") * Value.var("bp1")
+    + Value.var("q5") * Value.var("bp2")
+    + Value.var("q8") * Value.var("bp3")
+    + Value.var("bp4") * Value.var("q11")
+)
+_expr_2 = (
+    Value.var("bpp2") * Value.var("q6")
+    + Value.var("q3") * Value.var("bpp1")
+    + Value.var("bpp3") * Value.var("q9")
+    + Value.var("bpp4") * Value.var("q12")
+)
+_expr_3 = (
+    Value.var("bp1") * Value.var("q3")
+    + Value.var("bp2") * Value.var("q6")
+    + Value.var("q12") * Value.var("bp4")
+    + Value.var("bp3") * Value.var("q9")
+)
+_expr_4 = (
+    Value.var("bpp2") * Value.var("q5")
+    + Value.var("q2") * Value.var("bpp1")
+    + Value.var("bpp3") * Value.var("q8")
+    + Value.var("bpp4") * Value.var("q11")
+)
+_expr_5 = (
+    Value.var("q4") * Value.var("bpp2")
+    + Value.var("q7") * Value.var("bpp3")
+    + Value.var("q10") * Value.var("bpp4")
+    + Value.var("bpp1") * Value.var("q1")
+)
+_expr_6 = (
+    Value.var("q4") * Value.var("bp2")
+    + Value.var("q7") * Value.var("bp3")
+    + Value.var("bp1") * Value.var("q1")
+    + Value.var("q10") * Value.var("bp4")
+)
 NDArray(
     RecursiveValue(
         (
             (_expr_1 * _expr_2 + Value.from_int(Int(-1)) * (_expr_3 * _expr_4)) ** Value.from_int(Int(2))
             + (_expr_3 * _expr_5 + Value.from_int(Int(-1)) * (_expr_6 * _expr_2)) ** Value.from_int(Int(2))
             + (_expr_6 * _expr_4 + Value.from_int(Int(-1)) * (_expr_1 * _expr_5)) ** Value.from_int(Int(2))
         )
-        / (_expr_6 ** Value.from_int(Int(2)) + _expr_1 ** Value.from_int(Int(2)) + _expr_3 ** Value.from_int(Int(2))) ** Value.from_int(Int(3))
+        / (_expr_6 ** Value.from_int(Int(2)) + _expr_1 ** Value.from_int(Int(2)) + _expr_3 ** Value.from_int(Int(2)))
+        ** Value.from_int(Int(3))
     )
-)
+)

python/egglog/exp/param_eq/__main__.py

@@ -4,6 +4,5 @@
 
 from .pipeline import _cli
 
-
 if __name__ == "__main__":
     _cli()