From 28ceb41e1cdb603c4e0bd2fe7b72acef1b29ec47 Mon Sep 17 00:00:00 2001
From: you-n-g <you-n-g@users.noreply.github.com>
Date: Thu, 31 Jul 2025 18:10:32 +0800
Subject: [PATCH] fix: clear ws_ckp after extraction to reduce workspace object
 size (#1137)

---
 rdagent/core/experiment.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/rdagent/core/experiment.py b/rdagent/core/experiment.py
index 78a7cbed..337bd0f9 100644
--- a/rdagent/core/experiment.py
+++ b/rdagent/core/experiment.py
@@ -350,6 +350,8 @@ class FBWorkspace(Workspace):
                         dest_path.parent.mkdir(parents=True, exist_ok=True)
                         with dest_path.open("wb") as f:
                             f.write(zf.read(info))
+        # NOTE: very important to reduce the size of the object
+        self.ws_ckp = None
 
     def __str__(self) -> str:
         return f"Workspace[{self.workspace_path=}" + (