mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
Replace the _clear_queue function with celery_queue.clear
This commit is contained in:
committed by
Peter Rowlands (변기호)
parent
704191b15a
commit
a6063307b5
@@ -18,7 +18,7 @@ from .utils import (
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from dvc.repo import Repo
|
||||
from dvc.repo.experiments.queue.local import LocalCeleryQueue
|
||||
from dvc.repo.experiments.queue.celery import LocalCeleryQueue
|
||||
from dvc.scm import Git
|
||||
|
||||
|
||||
@@ -41,7 +41,7 @@ def remove(
|
||||
return removed
|
||||
|
||||
if queue:
|
||||
removed.extend(_clear_queue(repo))
|
||||
removed.extend(repo.experiments.celery_queue.clear(queued=True))
|
||||
if all_commits:
|
||||
removed.extend(_clear_all_commits(repo, git_remote))
|
||||
return removed
|
||||
@@ -117,14 +117,6 @@ def _resolve_exp_by_baseline(
|
||||
commit_ref_dict[ref_info] = ref_info.name
|
||||
|
||||
|
||||
def _clear_queue(repo: "Repo") -> List[str]:
|
||||
removed_name_list = []
|
||||
for entry in repo.experiments.celery_queue.iter_queued():
|
||||
removed_name_list.append(entry.name or entry.stash_rev[:7])
|
||||
repo.experiments.celery_queue.clear(queued=True)
|
||||
return removed_name_list
|
||||
|
||||
|
||||
def _clear_all_commits(repo, git_remote) -> List:
|
||||
ref_infos = {
|
||||
ref_info: ref_info.name for ref_info in exp_refs(repo.scm, git_remote)
|
||||
|
||||
Reference in New Issue
Block a user