Format saved parameter changes

Author: joshuawootonn

py/src/braintrust/cli/push.py

@@ -271,6 +271,7 @@ def _collect_prompt_function_defs(
     for p in global_.prompts:
         functions.append(p.to_function_definition(if_exists, project_ids))
 
+
 def _collect_evaluator_defs(
     project_ids: ProjectIdCache,
     functions: list[dict[str, Any]],

py/src/braintrust/parameters.py

@@ -10,6 +10,7 @@
 from .prompt import PromptData
 from .serializable_data_class import SerializableDataClass
 
+
 if TYPE_CHECKING:
     from .logger import Prompt
 

py/src/braintrust/test_framework.py

@@ -16,6 +16,7 @@
 from .score import Score, Scorer
 from .test_helpers import init_test_exp, with_memory_logger, with_simulate_login  # noqa: F401
 
+
 HAS_PYDANTIC = importlib.util.find_spec("pydantic") is not None
 
 

py/src/braintrust/test_logger.py

@@ -151,6 +151,7 @@ def test_init_with_saved_parameters_attaches_reference(self):
         assert payload["parameters_id"] == "params-123"
         assert payload["parameters_version"] == "v1"
 
+
 class TestLogger(TestCase):
     def test_load_parameters_returns_remote_object(self):
         mock_api_conn = MagicMock()
@@ -185,11 +186,14 @@ def test_load_parameters_returns_remote_object(self):
         assert parameters.id == "params-123"
         assert parameters.version == "v1"
         assert parameters.data == {"prefix": "hello"}
-        assert logger._state._parameters_cache.get(
-            slug="saved-parameters",
-            version="latest",
-            project_name="test-project",
-        ).id == "params-123"
+        assert (
+            logger._state._parameters_cache.get(
+                slug="saved-parameters",
+                version="latest",
+                project_name="test-project",
+            ).id
+            == "params-123"
+        )
 
     def test_extract_attachments_no_op(self):
         attachments: List[BaseAttachment] = []

py/src/braintrust/test_parameters.py

@@ -7,6 +7,7 @@
     validate_parameters,
 )
 
+
 HAS_PYDANTIC = importlib.util.find_spec("pydantic") is not None