formatting problem

Author: JasonHonKL

plexe/internal/common/utils/pydantic_utils.py

@@ -66,7 +66,7 @@ def map_to_basemodel(name: str, schema: dict | Type[BaseModel]) -> Type[BaseMode
         try:
             # Handle both Dict[str, type] and Dict[str, str] formats
             annotated_schema = {}
-            
+
             type_mapping = {
                 "int": int,
                 "float": float,

plexe/models.py

@@ -111,7 +111,9 @@ def __init__(
         # The model's identity is defined by these fields
         self.intent: str = intent
         self.input_schema: Optional[Type[BaseModel]] = map_to_basemodel("in", input_schema) if input_schema else None
-        self.output_schema: Optional[Type[BaseModel]] = map_to_basemodel("out", output_schema) if output_schema else None
+        self.output_schema: Optional[Type[BaseModel]] = (
+            map_to_basemodel("out", output_schema) if output_schema else None
+        )
         self.training_data: Dict[str, Dataset] = dict()
         self.distributed: bool = distributed