cleanup

Signed-off-by: Mandana Vaziri <[email protected]>

Author: vazirim

src/pdl/pdl_context.py

@@ -56,7 +56,7 @@ def __len__(self):  # pyright: ignore
         return 1
 
     def __getitem__(self, index: int | slice):  # pyright: ignore
-        if index == 0 or index == -1:
+        if index in (0, -1):
             return self.message.result()
         assert False
 

src/pdl/pdl_interpreter.py

@@ -95,8 +95,8 @@
     empty_block_location,
 )
 from .pdl_context import (  # noqa: E402
-    PDLContext,
     DependentContext,
+    PDLContext,
     SerializeMode,
     SingletonContext,
     add_done_callback,
@@ -367,7 +367,7 @@ def set_error_to_scope_for_retry(
         return scope
     if pdl_context:
         last_msg = pdl_context[-1]
-        last_error = last_msg["content"] # type: ignore
+        last_error = last_msg["content"]  # type: ignore
         if last_error.endswith(error):
             repeating_same_error = True
     if repeating_same_error:
@@ -377,9 +377,7 @@ def set_error_to_scope_for_retry(
         "content": error,
         "defsite": block_id,
     }
-    scope = scope | {
-        "pdl_context": pdl_context * SingletonContext(PdlDict(err_msg))
-    }
+    scope = scope | {"pdl_context": pdl_context * SingletonContext(PdlDict(err_msg))}
     return scope
 
 
@@ -418,7 +416,9 @@ def process_advanced_block(
                 state, scope, block, loc
             )
             result = lazy_apply(id_with_set_first_use_nanos(block.pdl__timing), result)
-            add_done_callback(id_with_set_first_use_nanos(block.pdl__timing), background)
+            add_done_callback(
+                id_with_set_first_use_nanos(block.pdl__timing), background
+            )
             trace = trace.model_copy(update={"pdl__result": result})
             if block.parser is not None:
                 # Use partial to create a function with fixed arguments