Formatting and typing

Signed-off-by: Louis Mandel <[email protected]>

Author: mandel

src/pdl/pdl_context.py

@@ -135,6 +135,7 @@ def __repr__(self):  # pyright: ignore
         ret += ",".join([i.__repr__() for i in self.context.result()])
         return ret + "]"
 
+
 def ensure_context(context: dict | list | PDLContext) -> PDLContext:
     match context:
         case dict():
@@ -147,6 +148,7 @@ def ensure_context(context: dict | list | PDLContext) -> PDLContext:
             raise TypeError(f"'{type(context)}' object is not a valid context")
     return ctx
 
+
 def deserialize(
     context: list[dict[str, Any]],
 ) -> DependentContext:  # Only support dependent for now

src/pdl/pdl_interpreter.py

@@ -1429,7 +1429,9 @@ def process_call_model(
                 "pdl__model_input": model_input,
             }
         )
-        model_input = [{k: v for k, v in m.items() if k != "defsite"} for m in model_input]
+        model_input = [
+            {k: v for k, v in m.items() if k != "defsite"} for m in model_input
+        ]
 
         # Execute model call
         litellm_params = {}