Apply black formatting to fix linting issues

Co-Authored-By: [email protected] <[email protected]>

examples/poem-generation/poem.py

@@ -40,13 +40,14 @@ class Poems(BaseModel):
 # Type alias for input/output types
 _DictOrBaseModel = Union[Dict[str, Any], BaseModel]
 
+
 def parse_poems(row: _DictOrBaseModel, poems: _DictOrBaseModel) -> _DictOrBaseModel:
     """Parse the poems from the LLM response.
-    
+
     Args:
         row: The input row containing the topic
         poems: The structured output from the LLM (Poems model)
-    
+
     Returns:
         A list of dictionaries containing the topic and poem
     """

src/bespokelabs/curator/llm/llm.py

@@ -284,7 +284,7 @@ def __call__(
 
 def _get_function_hash(func) -> str:
     """Get a hash of a function's source code.
-    
+
     Uses cloudpickle to properly handle functions with type annotations and closure variables.
     """
     if func is None:

tests/test_prompt.py

@@ -98,10 +98,10 @@ def test_single_completion_batch(prompter: LLM):
     # Create a prompter with batch=True
     def simple_prompt_func(row: _DictOrBaseModel) -> _DictOrBaseModel:
         """Generate a simple prompt for testing.
-        
+
         Args:
             row: The input row (unused in this test)
-            
+
         Returns:
             A list of messages for the LLM
         """
@@ -155,10 +155,10 @@ def test_single_completion_no_batch(prompter: LLM):
     # Create a prompter without batch parameter
     def simple_prompt_func(row: _DictOrBaseModel) -> _DictOrBaseModel:
         """Generate a simple prompt for testing.
-        
+
         Args:
             row: The input row (unused in this test)
-            
+
         Returns:
             A list of messages for the LLM
         """