Format with ruff

Author: tgasser-nv

nemoguardrails/colang/v1_0/lang/coyml_parser.py

@@ -420,12 +420,8 @@ def _extract_elements(items: List) -> List[dict]:
             # for `if` flow elements, we have to go recursively
             if element["_type"] == "if":
                 if_element = element
-                then_items = (
-                    if_element["then"] if isinstance(if_element["then"], list) else []
-                )
-                else_items = (
-                    if_element["else"] if isinstance(if_element["else"], list) else []
-                )
+                then_items = if_element["then"] if isinstance(if_element["then"], list) else []
+                else_items = if_element["else"] if isinstance(if_element["else"], list) else []
                 then_elements = _extract_elements(then_items)
                 else_elements = _extract_elements(else_items)
 
@@ -443,9 +439,7 @@ def _extract_elements(items: List) -> List[dict]:
 
                 # if we have "else" elements, we need to adjust also add a jump
                 if len(else_elements) > 0:
-                    elements.append(
-                        {"_type": "jump", "_next": str(len(else_elements) + 1)}
-                    )
+                    elements.append({"_type": "jump", "_next": str(len(else_elements) + 1)})
                     if_element["_next_else"] = str(int(if_element["_next_else"]) + 1)
 
                     # Add the "else" elements
@@ -454,9 +448,7 @@ def _extract_elements(items: List) -> List[dict]:
             # WHILE
             elif element["_type"] == "while":
                 while_element = element
-                do_items = (
-                    while_element["do"] if isinstance(while_element["do"], list) else []
-                )
+                do_items = while_element["do"] if isinstance(while_element["do"], list) else []
                 do_elements = _extract_elements(do_items)
                 n = len(do_elements)
 

nemoguardrails/colang/v2_x/lang/transformer.py

@@ -174,9 +174,7 @@ def _flow_def(self, children: dict, meta: Meta) -> Flow:
                 if len(return_member_def["elements"]) == 2:
                     default_value_el = return_member_def["elements"][1]
                     assert default_value_el["_type"] == "expr"
-                    return_member_def_obj.default_value_expr = default_value_el[
-                        "elements"
-                    ][0]
+                    return_member_def_obj.default_value_expr = default_value_el["elements"][0]
 
                 return_member_defs.append(return_member_def_obj)
 

nemoguardrails/logging/processing_log.py

@@ -23,9 +23,9 @@
 )
 
 # The processing log for the current async stack
-processing_log_var: contextvars.ContextVar[
-    Optional[List[dict]]
-] = contextvars.ContextVar("processing_log", default=None)
+processing_log_var: contextvars.ContextVar[Optional[List[dict]]] = contextvars.ContextVar(
+    "processing_log", default=None
+)
 
 
 def compute_generation_log(processing_log: List[dict]) -> GenerationLog: