From 476cc90ef77ca2d239951d47bea8c70df34e1236 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Thu, 28 Dec 2023 15:10:47 -0800
Subject: [PATCH] Lint

---
 permchain/pregel/__init__.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/permchain/pregel/__init__.py b/permchain/pregel/__init__.py
index c2b8519a6..31595be4d 100644
--- a/permchain/pregel/__init__.py
+++ b/permchain/pregel/__init__.py
@@ -214,9 +214,11 @@ class Pregel(RunnableSerializable[dict[str, Any] | Any, dict[str, Any] | Any]):
         # get checkpoint from saver, or create an empty one
         checkpoint = self.saver.get(config) if self.saver else None
         checkpoint = checkpoint or empty_checkpoint()
+        # create channels from checkpoint
         with ChannelsManager(
             self.channels, checkpoint
         ) as channels, get_executor_for_config(config) as executor:
+            # map inputs to channel updates
             _apply_writes(
                 checkpoint,
                 channels,
@@ -309,7 +311,9 @@ class Pregel(RunnableSerializable[dict[str, Any] | Any, dict[str, Any] | Any]):
         # get checkpoint from saver, or create an empty one
         checkpoint = await self.saver.aget(config) if self.saver else None
         checkpoint = checkpoint or empty_checkpoint()
+        # create channels from checkpoint
         async with AsyncChannelsManager(self.channels, checkpoint) as channels:
+            # map inputs to channel updates
             _apply_writes(
                 checkpoint,
                 channels,