Merge pull request #29 from mohamadkhalaj/main

Remove walrus
Aggify · Nov 1, 2023 · 5e77408 · 5e77408
2 parents 164e6e1 + 2731f84
commit 5e77408
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/aggify/aggify.py b/aggify/aggify.py
@@ -34,7 +34,7 @@ def last_out_stage_check(method):
     @functools.wraps(method)
     def decorator(*args, **kwargs):
         try:
-            if last_is_out := bool(args[0].pipelines[-1].get("$out")):
+            if bool(args[0].pipelines[-1].get("$out")):
                 raise OutStageError(method.__name__)
         except IndexError:
             return method(*args, **kwargs)

diff --git a/aggify/compiler.py b/aggify/compiler.py
@@ -230,7 +230,8 @@ def __init__(self, matches: dict[str, Any], base_model: Type[Document] | None):
 
     @staticmethod
     def validate_operator(key: str):
-        if (operator := key.rsplit("__", 1)[1]) not in Operators.COMPARISON_OPERATORS:
+        operator = key.rsplit("__", 1)[1]
+        if operator not in Operators.COMPARISON_OPERATORS:
             raise InvalidOperator(operator)
 
     def is_base_model_field(self, field) -> bool: