formatting

python/langsmith/anonymizer.py

@@ -92,12 +92,16 @@ def __init__(self, rules: List[StringNodeRule]):
         """Initialize the processor with a list of rules."""
         self.rules = [
             {
-                "pattern": rule["pattern"]
-                if isinstance(rule["pattern"], re.Pattern)
-                else re.compile(rule["pattern"]),
-                "replace": rule["replace"]
-                if isinstance(rule.get("replace"), str)
-                else "[redacted]",
+                "pattern": (
+                    rule["pattern"]
+                    if isinstance(rule["pattern"], re.Pattern)
+                    else re.compile(rule["pattern"])
+                ),
+                "replace": (
+                    rule["replace"]
+                    if isinstance(rule.get("replace"), str)
+                    else "[redacted]"
+                ),
             }
             for rule in rules
         ]

python/tests/unit_tests/test_anonymizer.py

@@ -140,16 +140,17 @@ def my_func(body: str, from_: MyInput) -> MyOutput:
         assert patched_data["inputs"] == expected_inputs
     assert patched_data["outputs"] == expected_outputs
 
+
 def test_rule_node_processor_scrub_sensitive_info():
     rules = [
-        StringNodeRule(pattern=re.compile(r"\b\d{3}-\d{2}-\d{4}\b"), replace="[ssn]"), 
+        StringNodeRule(pattern=re.compile(r"\b\d{3}-\d{2}-\d{4}\b"), replace="[ssn]"),
         StringNodeRule(
             pattern=re.compile(r"[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}"),
             replace="[email]",
-        ),  
+        ),
         StringNodeRule(
             pattern=re.compile(r"\b\d{3}[-.\s]?\d{3}[-.\s]?\d{4}\b"), replace="[phone]"
-        ), 
+        ),
     ]
     processor = RuleNodeProcessor(rules)
 
@@ -169,6 +170,7 @@ def test_rule_node_processor_scrub_sensitive_info():
 
     assert result == expected
 
+
 def test_rule_node_processor_default_replace():
     rules = [
         StringNodeRule(pattern=re.compile(r"sensitive")),
@@ -184,4 +186,4 @@ def test_rule_node_processor_default_replace():
     ]
 
     result = processor.mask_nodes(nodes)
-    assert result == expected
+    assert result == expected