From c16908e9fc0a8342fbb5c2e80d43e935bf29aea4 Mon Sep 17 00:00:00 2001 From: Peter Rowlands Date: Tue, 26 Apr 2022 16:54:48 +0900 Subject: [PATCH] exp: use queue remove in gc --- dvc/repo/experiments/gc.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/dvc/repo/experiments/gc.py b/dvc/repo/experiments/gc.py index 666d31f65..edb57a1e3 100644 --- a/dvc/repo/experiments/gc.py +++ b/dvc/repo/experiments/gc.py @@ -42,11 +42,12 @@ def gc( removed = len(to_remove) delete_stashes = [] - for _, entry in repo.experiments.workspace_queue.stash.stash_revs.items(): + stash = repo.experiments.celery_queue.stash + for stash_rev, entry in stash.stash_revs.items(): if not queued or entry.baseline_rev not in keep_revs: - delete_stashes.append(entry.stash_index) - for index in sorted(delete_stashes, reverse=True): - repo.experiments.workspace_queue.stash.drop(index) + delete_stashes.append(stash_rev) + if delete_stashes: + repo.experiments.celery_queue.remove(delete_stashes) removed += len(delete_stashes) return removed