From f0b7fa2466cb92f12f4a78256a3cc451de29c05d Mon Sep 17 00:00:00 2001 From: you-n-g Date: Sat, 30 Aug 2025 07:27:12 +0800 Subject: [PATCH] fix: allow prev_out keys to be None in workspace cleanup assertion (#1214) * fix: allow prev_out keys to be None in workspace cleanup assertion * fix: clean workspace only for non-None DSExperiment instances --- rdagent/scenarios/data_science/loop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rdagent/scenarios/data_science/loop.py b/rdagent/scenarios/data_science/loop.py index dffe173c..4738996f 100644 --- a/rdagent/scenarios/data_science/loop.py +++ b/rdagent/scenarios/data_science/loop.py @@ -311,7 +311,7 @@ class DataScienceRDLoop(RDLoop): # only clean current workspace without affecting other loops. for k in "direct_exp_gen", "coding", "running": - if k in prev_out: + if k in prev_out and prev_out[k] is not None: assert isinstance(prev_out[k], DSExperiment) clean_workspace(prev_out[k].experiment_workspace.workspace_path)