From eae4e789fe052de2835d4eacca3c0f81b5a31fe5 Mon Sep 17 00:00:00 2001
From: bojiang <5886138+bojiang@users.noreply.github.com>
Date: Tue, 21 Jan 2025 19:43:33 +0800
Subject: [PATCH] fix: undefined variable (#28)

---
 src/comfy_pack/package.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/comfy_pack/package.py b/src/comfy_pack/package.py
index 2bf5f4b..6c5bd05 100644
--- a/src/comfy_pack/package.py
+++ b/src/comfy_pack/package.py
@@ -456,11 +456,11 @@ def build_bento(
         system_packages = system_packages or []
 
     shutil.copy2(Path(__file__).with_name("service.py"), source_dir / "service.py")
+    snapshot_text = (source_dir / "snapshot.json").read_text()
     with Path(__file__).with_name("setup_workspace.py").open() as f:
         (source_dir / "setup_workspace.py").write_text(
             f.read().format(snapshot=snapshot_text)
         )
-    snapshot_text = (source_dir / "snapshot.json").read_text()
     snapshot = json.loads(snapshot_text)
     return bentoml.build(
         "service:ComfyService",