fix: minor fixes

Author: f4roukb

libs/langgraph-checkpoint-aws/langgraph_checkpoint_aws/buffered_saver.py

@@ -12,7 +12,6 @@
     CheckpointTuple,
 )
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -96,7 +95,7 @@ def __init__(
         """
         if isinstance(saver, BufferedCheckpointSaver):
             raise ValueError(f"{type(self).__name__} cannot be nested")
-        
+
         super().__init__(serde=saver.serde)
 
         self._saver = saver
@@ -145,6 +144,7 @@ def flush_on_exit(self):
             # Checkpoint and writes are automatically flushed on exit
             ```
         """
+
         def try_flush():
             try:
                 self.flush()
@@ -153,7 +153,7 @@ def try_flush():
                     "Failed to auto-flush buffered checkpoint and writes on exit."
                 )
                 raise
-        
+
         try:
             yield self
         except Exception:
@@ -180,6 +180,7 @@ async def aflush_on_exit(self):
             # Checkpoint and writes are automatically flushed on exit
             ```
         """
+
         async def try_aflush():
             try:
                 await self.aflush()
@@ -188,7 +189,7 @@ async def try_aflush():
                     "Failed to auto-flush buffered checkpoint and writes on exit."
                 )
                 raise
-        
+
         try:
             yield self
         except Exception:

libs/langgraph-checkpoint-aws/tests/integration_tests/agentcore/conftest.py

@@ -32,12 +32,11 @@ def agentcore_actor_id() -> str:
 
 @pytest.fixture(scope="function")
 def agentcore_memory_saver(
-        agentcore_memory_id,
-        aws_region,
-    ) -> AgentCoreMemorySaver:
+    agentcore_memory_id,
+    aws_region,
+) -> AgentCoreMemorySaver:
     """Create AgentCoreMemorySaver instance."""
     return AgentCoreMemorySaver(
         memory_id=agentcore_memory_id,
         region_name=aws_region,
     )
-