|
142 | 142 | } |
143 | 143 |
|
144 | 144 |
|
145 | | -def check_eq(x: BASE_EXPR, y: BASE_EXPR, schedule: Schedule | None = None, *actions: ActionLike, display=False) -> EGraph: |
| 145 | +def check_eq( |
| 146 | + x: BASE_EXPR, y: BASE_EXPR, schedule: Schedule | None = None, *actions: ActionLike, display=False |
| 147 | +) -> EGraph: |
146 | 148 | """ |
147 | 149 | Verifies that two expressions are equal after running the schedule. |
148 | 150 |
|
@@ -478,9 +480,7 @@ def _generate_class_decls( # noqa: C901,PLR0912 |
478 | 480 | subsume=False, |
479 | 481 | ) |
480 | 482 | resolved_default = ( |
481 | | - resolve_literal(type_ref, default_value, Thunk.value(decls)) |
482 | | - if default_mode == "eager" |
483 | | - else None |
| 483 | + resolve_literal(type_ref, default_value, Thunk.value(decls)) if default_mode == "eager" else None |
484 | 484 | ) |
485 | 485 | if resolved_default is not None: |
486 | 486 | decls |= resolved_default |
@@ -911,11 +911,7 @@ def _constant_thunk( |
911 | 911 | unextractable=False, |
912 | 912 | subsume=False, |
913 | 913 | ) |
914 | | - resolved_default = ( |
915 | | - resolve_literal(type_ref, default_replacement, Thunk.value(decls)) |
916 | | - if mode == "eager" |
917 | | - else None |
918 | | - ) |
| 914 | + resolved_default = resolve_literal(type_ref, default_replacement, Thunk.value(decls)) if mode == "eager" else None |
919 | 915 | if resolved_default is not None: |
920 | 916 | decls |= resolved_default |
921 | 917 | decls._constants[ident] = ConstantDecl( |
@@ -1582,11 +1578,7 @@ def all_function_sizes(self) -> list[tuple[ExprCallable, int]]: |
1582 | 1578 | """ |
1583 | 1579 | (output,) = self._state.run_program(bindings.PrintSize(span(1), None)) |
1584 | 1580 | assert isinstance(output, bindings.PrintAllFunctionsSize) |
1585 | | - return [ |
1586 | | - (callables[0], size) |
1587 | | - for (name, size) in output.sizes |
1588 | | - if (callables := self._egg_fn_to_callables(name)) |
1589 | | - ] |
| 1581 | + return [(callables[0], size) for (name, size) in output.sizes if (callables := self._egg_fn_to_callables(name))] |
1590 | 1582 |
|
1591 | 1583 | def _egg_fn_to_callables(self, egg_fn: str) -> list[ExprCallable]: |
1592 | 1584 | return [ |
|
0 commit comments